mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 22:33:57 +01:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
|
import os
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from dotenv import load_dotenv
|
||
|
|
from motor.motor_asyncio import AsyncIOMotorClient
|
||
|
|
|
||
|
|
load_dotenv()
|
||
|
|
|
||
|
|
class FinanceManager:
|
||
|
|
def __init__(self):
|
||
|
|
self.conn = AsyncIOMotorClient(os.environ['MONGO_URI'])
|
||
|
|
|
||
|
|
async def _get_collection(self, collection_name: str):
|
||
|
|
return self.conn['finances'][collection_name]
|
||
|
|
|
||
|
|
async def get_entire_financial_history(self):
|
||
|
|
donations_db = await self._get_collection('donations')
|
||
|
|
expenses_db = await self._get_collection('expenses')
|
||
|
|
|
||
|
|
# turn both into JSON-like lists of dicts at once (make sure to fix the _id)
|
||
|
|
history = {'donations': [], 'expenses': []}
|
||
|
|
|
||
|
|
async for donation in donations_db.find():
|
||
|
|
donation['_id'] = str(donation['_id'])
|
||
|
|
history['donations'].append(donation)
|
||
|
|
|
||
|
|
async for expense in expenses_db.find():
|
||
|
|
expense['_id'] = str(expense['_id'])
|
||
|
|
history['expenses'].append(expense)
|
||
|
|
|
||
|
|
# sort all by timestamp
|
||
|
|
history['donations'] = sorted(history['donations'], key=lambda x: x['timestamp'])
|
||
|
|
|
||
|
|
return history
|
||
|
|
|
||
|
|
manager = FinanceManager()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
print(asyncio.run(manager.get_entire_financial_history()))
|