mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-29 14:23:57 +01:00
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
import os
|
|
import time
|
|
|
|
from dotenv import load_dotenv
|
|
from motor.motor_asyncio import AsyncIOMotorClient
|
|
|
|
from helpers import network
|
|
|
|
load_dotenv()
|
|
|
|
UA_SIMPLIFY = {
|
|
'Windows NT': 'W',
|
|
'Mozilla/5.0': 'M',
|
|
'Win64; x64': '64',
|
|
'Safari/537.36': 'S',
|
|
'AppleWebKit/537.36 (KHTML, like Gecko)': 'K',
|
|
}
|
|
|
|
## MONGODB Setup
|
|
|
|
conn = AsyncIOMotorClient(os.getenv('MONGO_URI'))
|
|
|
|
async def _get_collection(collection_name: str):
|
|
return conn[os.getenv('MONGO_NAME', 'nova-test')][collection_name]
|
|
|
|
async def replacer(text: str, dict_: dict) -> str:
|
|
# This seems to exist for a very specific and dumb purpose :D
|
|
for k, v in dict_.items():
|
|
text = text.replace(k, v)
|
|
return text
|
|
|
|
async def log_api_request(user: dict, incoming_request, target_url: str):
|
|
"""Logs the API Request into the database.
|
|
No input prompt is logged, however data such as IP & useragent is noted.
|
|
This would be useful for security reasons. Other minor data is also collected.
|
|
|
|
Args:
|
|
user (dict): User dict object
|
|
incoming_request (_type_): Request
|
|
target_url (str): The URL the api request was targetted to.
|
|
|
|
Returns:
|
|
_type_: _description_
|
|
"""
|
|
db = await _get_collection('logs')
|
|
payload = {}
|
|
|
|
try:
|
|
payload = await incoming_request.json()
|
|
except Exception as exc:
|
|
if 'JSONDecodeError' in str(exc):
|
|
pass
|
|
|
|
model = payload.get('model')
|
|
ip_address = await network.get_ip(incoming_request)
|
|
useragent = await replacer(incoming_request.headers.get('User-Agent'), UA_SIMPLIFY)
|
|
|
|
new_log_item = {
|
|
'timestamp': time.time(),
|
|
'method': incoming_request.method,
|
|
'path': incoming_request.url.path,
|
|
'user_id': str(user['_id']),
|
|
'security': {
|
|
'ip': ip_address,
|
|
'useragent': useragent,
|
|
},
|
|
'details': {
|
|
'model': model,
|
|
'target_url': target_url
|
|
}
|
|
}
|
|
|
|
inserted = await db.insert_one(new_log_item)
|
|
log_item = await db.find_one({'_id': inserted.inserted_id})
|
|
return log_item
|
|
|
|
async def by_id(log_id: str):
|
|
db = await _get_collection('logs')
|
|
return await db.find_one({'_id': log_id})
|
|
|
|
async def by_user_id(user_id: str):
|
|
db = await _get_collection('logs')
|
|
return await db.find({'user_id': user_id})
|
|
|
|
async def delete_by_id(log_id: str):
|
|
db = await _get_collection('logs')
|
|
return await db.delete_one({'_id': log_id})
|
|
|
|
async def delete_by_user_id(user_id: str):
|
|
db = await _get_collection('logs')
|
|
return await db.delete_many({'user_id': user_id})
|
|
|
|
if __name__ == '__main__':
|
|
pass
|