fix final step when users are dictionaries

datawhores · Aug 21, 2024 · 5cf9a23 · 5cf9a23
1 parent 8d651bf
commit 5cf9a23
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ofscraper/runner/close/final/final_script.py b/ofscraper/runner/close/final/final_script.py
@@ -31,7 +31,9 @@ def final_script(users):
     log.debug("Running post script")
     data = {}
     for ele in users:
-        data[ele.id] = ele.model if isinstance(ele, Model) else ele
+        key=ele.model if isinstance(ele, Model) else ele["id"]
+        value=ele.model if isinstance(ele, Model) else ele
+        data[key] = value
     out_dict = json.dumps(
         {
             "users": data,