feat(proxy_server.py): adds working dynamo db support for key gen

This commit is contained in:
Krrish Dholakia 2024-01-09 18:11:24 +05:30
parent b09f38e835
commit 00c258c165
5 changed files with 362 additions and 30 deletions

View file

@ -0,0 +1,43 @@
from typing import Any, Literal, List
class CustomDB:
"""
Implements a base class that we expect any custom db implementation (e.g. DynamoDB) to follow
"""
def __init__(self) -> None:
pass
def get_data(self, key: str, value: str, table_name: Literal["user", "key", "config"]):
"""
Check if key valid
"""
pass
def insert_data(self, value: Any, table_name: Literal["user", "key", "config"]):
"""
For new key / user logic
"""
pass
def update_data(self, key: str, value: Any, table_name: Literal["user", "key", "config"]):
"""
For cost tracking logic
"""
pass
def delete_data(self, keys: List[str], table_name: Literal["user", "key", "config"]):
"""
For /key/delete endpoint s
"""
def connect(self, ):
"""
For connecting to db and creating / updating any tables
"""
pass
def disconnect(self, ):
"""
For closing connection on server shutdown
"""
pass