bug fix writing data to excel

arxml_data_extractor/data_writer.py

@@ -59,8 +59,10 @@ def write_excel(self, file: str, data: dict, queries: List[DataObject]):
 
         workbook.close()
 
-    def __write_worksheet(self, sheet, data: dict, query: DataObject):
+    def __write_worksheet(self, sheet, data: Union[dict, list], query: DataObject):
         header = self.__extract_header(query)
+        if isinstance(data, dict):
+            data = [data]
         df = self.__flatten(data)
 
         start = self.__write_header(sheet, header)
@@ -122,10 +124,10 @@ def __flatten(cls, data: Union[list, dict]):
                 if isinstance(value, dict):
                     rows.extend(cls.__flatten(value))
                 elif isinstance(value, list):
-                    row = rows.copy()
+                    copy = rows.copy()
                     rows.clear()
                     for v in value:
-                        rows.append(row + cls.__flatten(v))
+                        rows.append(copy + cls.__flatten(v))
                 else:
                     rows.append(value)
         elif isinstance(data, list):