cleanup and slight refactor

main.py

@@ -1,22 +1,35 @@
-from vrcapi import getGroupInstances, fakeGroupInstances
+import os
+import logging
+from vrcapi import get_group_instances
+from testing import fake_group_instances
 from datetime import datetime
 from db import DBGroupInstance, get_session
 
+log_level = os.getenv("LOG_LEVEL")
+logging.basicConfig(level=log_level)
 session = get_session()
 
+# Commit the group instance data in the DB
+def store_instance_data(instance):
+    try:
+        db_instance = DBGroupInstance(
+            timestamp=datetime.now(),
+            instance_id=instance.instance_id,
+            world_name=instance.world.name,
+            member_count=instance.member_count,
+            delivered=False # Discord delivery, always false until separate job marks it as True
+        )
+        session.add(db_instance)
+        session.commit()
+        logging.info("Instance log added")
+    except Exception as e:
+        logging.error(f"Exception while adding instance log: {e}")
+        session.rollback()
+
 if __name__ == "__main__":
-    group_instances = getGroupInstances()
+    group_instances = get_group_instances() # Call the VRChat API and pull group instance data
     if not group_instances:
-        print("No group instances were found! Nothing to do...")
+        logging.info("No group instances were found! Nothing to do...") # Since this will run in a cron job, the program ends here
     else:
         for instance in group_instances:
-            try:
-                db_instance = DBGroupInstance(timestamp=datetime.now(),
-                                              instance_id=instance.instance_id,
-                                              world_name=instance.world.name,
-                                              member_count=instance.member_count)
-                session.add(db_instance)
-                session.commit()
-                print("Instance log added")
-            except Exception as e:
-                print(f"Exception {e}")
+            store_instance_data(instance) # Send instance data to the DB