Add asyncpg logic

hexfrost · Jan 8, 2025 · 989e81d · 989e81d
1 parent 75abae3
commit 989e81d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -7,4 +7,4 @@ package-install:
 lint:
 	poetry run flake8 simplecrud
 coverage:
-	poetry run coverage run --source=simplecrud -m pytest
+	poetry run pytest --cov=simplecrud --cov-report=xml
diff --git a/simplecrud/sync_crud.py → simplecrud/asyncpg_commands/__init__.py b/simplecrud/sync_crud.py → simplecrud/asyncpg_commands/__init__.py
diff --git a/simplecrud/asyncpg_commands/cruds.py b/simplecrud/asyncpg_commands/cruds.py
@@ -0,0 +1,10 @@
+from asyncpg import connect
+
+
+async def insert_to_table(conn: connect, table: str, data: dict):
+    keys = ", ".join(data.keys())
+    nums = ", ".join([f"${num}" for num in range(1, len(data.items()) + 1)])
+    values = list(data.values())
+    query = f"INSERT INTO {table} ({keys}) VALUES ({nums}) ON CONFLICT DO NOTHING;"
+    await conn.execute(query, *values)
+