xevion/postgres-test
1
0
Fork 0
mirror of https://github.com/Xevion/postgres-test.git synced 2025-12-06 09:15:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add SQL functions for mass insert, rename cursor type to Cursor for type brevity

Browse Source
This commit is contained in:
Xevion
2023-04-23 14:30:43 -05:00
parent f14f20336f
commit 2da765fa31
1 changed files with 46 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/sql.py
View File

@@ -1,9 +1,11 @@
from psycopg2._psycopg import cursor
from typing import List
from types import Account, Posting, PostingFilter
from psycopg2._psycopg import cursor as Cursor
from src.types import Account, Posting, PostingFilter
def add_account(cursor: cursor, data: Account) -> int:
def add_account(cursor: Cursor, data: Account) -> int:
"""Adds an account to the database."""
cursor.execute('INSERT INTO Account'
'(first_name, last_name, email)'
@@ -13,7 +15,17 @@ def add_account(cursor: cursor, data: Account) -> int:
return cursor.fetchone()[0]
def add_posting(cursor: cursor, data: Posting) -> int:
def add_accounts(cursor: Cursor, data: List[Account]) -> List[int]:
"""Adds multiple accounts to the database."""
cursor.executemany('INSERT INTO Account'
'(first_name, last_name, email)'
'VALUES (%s, %s, %s)'
'RETURNING id',
[(account.first_name, account.last_name, account.email) for account in data])
return [id for (id,) in cursor.fetchall()]
def add_posting(cursor: Cursor, data: Posting) -> int:
"""Adds a posting to the database."""
cursor.execute('INSERT INTO Posting'
'(name, date, description, a, b, c)'
@@ -23,7 +35,18 @@ def add_posting(cursor: cursor, data: Posting) -> int:
return cursor.fetchone()[0]
def add_posting_filter(cursor: cursor, data: PostingFilter) -> int:
def add_postings(cursor: Cursor, data: List[Posting]) -> List[int]:
"""Adds multiple postings to the database."""
cursor.executemany('INSERT INTO Posting'
'(name, date, description, a, b, c)'
'VALUES (%s, %s, %s, %s, %s, %s)'
'RETURNING id',
[(posting.name, posting.date, posting.description, posting.a, posting.b, posting.c) for posting
in data])
return [id for (id,) in cursor.fetchall()]
def add_posting_filter(cursor: Cursor, data: PostingFilter) -> int:
"""Adds a posting filter to the database."""
b_lower, b_upper = data.b if data.b else (None, None)
c_lower, c_upper = data.c if data.c else (None, None)
@@ -33,3 +56,21 @@ def add_posting_filter(cursor: cursor, data: PostingFilter) -> int:
'RETURNING id',
(data.creator, data.name, data.expires, data.a, b_lower, b_upper, c_lower, c_upper))
return cursor.fetchone()[0]
def add_posting_filters(cursor: Cursor, data: List[PostingFilter]) -> List[int]:
"""Adds multiple posting filters to the database."""
cursor.executemany('INSERT INTO PostingFilter'
'(creator, name, expires, a, b_lower, b_upper, c_lower, c_upper)'
'VALUES (%s, %s, %s, %s, %s, %s)'
'RETURNING id',
[(posting_filter.creator, posting_filter.name, posting_filter.expires, posting_filter.a,
posting_filter.b[0], posting_filter.b[1], posting_filter.c[0], posting_filter.c[1]) for
posting_filter in data])
return [id for (id,) in cursor.fetchall()]
def get_account_ids(cur: Cursor):
"""Returns a list of all account IDs."""
cur.execute('SELECT id FROM Account')
return [id for (id,) in cur.fetchall()]