53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from datetime import datetime
|
|
from database import get_connection, is_postgres
|
|
|
|
class Currency:
|
|
def __init__(self):
|
|
self.ph = "%s" if is_postgres() else "?"
|
|
|
|
def currency_to_dict(self, row):
|
|
currency = {
|
|
'id': row[0],
|
|
'user_id': row[1],
|
|
'name': row[2],
|
|
'value': row[3],
|
|
'created_at': row[4]
|
|
}
|
|
return currency
|
|
|
|
def insert_currency(self, access_data):
|
|
created_at = datetime.now().isoformat()
|
|
with get_connection() as conn:
|
|
cursor = conn.cursor()
|
|
returning = "RETURNING id" if is_postgres() else ""
|
|
query = f"""
|
|
INSERT INTO currency (
|
|
user_id, name, value, created_at
|
|
) VALUES ({self.ph}, {self.ph}, {self.ph}, {self.ph}) {returning}
|
|
"""
|
|
cursor.execute(
|
|
query,
|
|
(
|
|
access_data['user_id'],
|
|
access_data['name'],
|
|
access_data['value'],
|
|
created_at
|
|
)
|
|
)
|
|
inserted_id = None
|
|
if is_postgres():
|
|
inserted_id = cursor.fetchone()[0]
|
|
else:
|
|
inserted_id = cursor.lastrowid
|
|
if hasattr(conn, "commit"):
|
|
conn.commit()
|
|
return inserted_id
|
|
|
|
def get_currency_user_id(self, user_id):
|
|
with get_connection() as conn:
|
|
cursor = conn.cursor()
|
|
cursor.execute(f"SELECT * FROM currency WHERE user_id = {self.ph}", (user_id,))
|
|
row = cursor.fetchone()
|
|
return self.currency_to_dict(row) if row else None
|
|
|
|
|