db support

This commit is contained in:
2024-06-12 14:33:07 +00:00
parent ef2513b43e
commit 78fb2f5fff
3 changed files with 48 additions and 11 deletions

28
main.py
View File

@@ -1,14 +1,22 @@
from vrcapi import getGroupInstances, fakeGroupInstances
from datetime import datetime
from db import DBGroupInstance, get_session
session = get_session()
if __name__ == "__main__":
instance_data = []
group_instances = fakeGroupInstances()
for instance in group_instances:
instance_record = {"timestamp": datetime.now(),
"instance_id": {instance.instance_id},
"member_count": {instance.member_count}}
instance_data.append(instance_record)
print(instance_data)
group_instances = getGroupInstances()
if group_instances is not []:
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}")
else:
print("No group instances! Nothing to do")