db support
This commit is contained in:
28
main.py
28
main.py
@@ -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")
|
Reference in New Issue
Block a user