Adicionar repositórios de usuário para MongoDB e PostgreSQL

Author: maxsonferovante

models/repository/__init__.py

@@ -0,0 +1,50 @@
+import asyncio
+
+from typing import AsyncGenerator
+
+class UserMongoRepository:
+    def __init__(self, user_collenction):
+        
+        self.__collenction = user_collenction
+
+    async def create_user(self, user: dict) -> int:
+        
+        result = await self.__collenction.insert_one(user)
+
+        return result.inserted_id
+    
+    async def get_user(self, user_id: int) -> dict:
+        user = await self.__collenction.find_one(
+            {"id": user_id}
+        )
+    
+    async def count_documents(self) -> int:
+        return await self.__collenction.count_documents({})
+    
+    async def create_users(self, users: list) -> None:
+        await self.__collenction.insert_many(users)
+        
+    async def get_all() -> list:
+        users_curson = self.__collenction.find().sort("id")
+        return await users_curson.to_list()
+    
+    async def get_all_paginated(self, skip: int, limit: int) -> AsyncGenerator[list, None]:
+        users_curson = self.__collenction.find().skip(skip).limit(limit)
+    
+        users_list = await users_curson.to_list(
+            length=limit
+        )
+        
+        if not users_list:
+            return
+        yield users_list
+        
+        async for more_users in self.get_all_paginated(skip + limit, limit):
+            yield more_users
+    
+    async def delete_all(self) -> None:
+        await self.__collenction.delete_many({})
+        
+
+    
+        

models/repository/user_mongo_repository.py

@@ -0,0 +1,31 @@
+import asyncio
+
+class UserPostgresRepository:
+    def __init__(self, conn):
+        self.conn = conn
+    
+    
+    async def create_table(self):
+        create_table_query = '''CREATE TABLE IF NOT EXISTS users (
+            id SERIAL PRIMARY KEY,
+            username VARCHAR(50) NOT NULL,
+            email VARCHAR(50) NOT NULL,
+            age INT NOT NULL,
+            registed_at TIMESTAMP NOT NULL DEFAULT NOW()        
+        )'''
+        await self.conn.fetch(create_table_query)
+    
+    async def insert_user(self, username, email, age):
+        insert_query = '''INSERT INTO users (username, email, age) VALUES ($1, $2, $3)'''
+        await self.conn.fetch(insert_query, username, email, age)
+    
+    async def get_users(self):
+        return await self.conn.fetch('''SELECT * FROM users''')
+    
+    async def drop_table(self):
+        await self.conn.fetch('''DROP TABLE IF EXISTS users''')
+        
+    async def count_users(self):
+        return await self.conn.fetchval('''SELECT COUNT(*) FROM users''')
+    
+