Luís Murta
ea3c75245a
The __conform__ can´t generate a tuple from the class, but it is still worth to use the DB intermediate types for cleaner code. So add tuple() method the the DBTransaction and use it when writing to the DB.
159 lines
2.8 KiB
Python
159 lines
2.8 KiB
Python
from dataclasses import dataclass
|
|
from decimal import Decimal
|
|
|
|
CREATE_TRANSACTIONS_TABLE = """
|
|
CREATE TABLE IF NOT EXISTS "transactions" (
|
|
"date" TEXT NOT NULL,
|
|
"description" TEXT,
|
|
"bank" TEXT NOT NULL,
|
|
"value" REAL NOT NULL,
|
|
"category" TEXT,
|
|
"original" TEXT,
|
|
"additional comments" TEXT
|
|
);
|
|
"""
|
|
|
|
|
|
@dataclass
|
|
class DbTransaction:
|
|
date: str
|
|
description: str
|
|
bank: str
|
|
value: Decimal
|
|
category: str
|
|
original: str
|
|
comments: str
|
|
|
|
def tuple(self) -> tuple:
|
|
return (
|
|
self.date,
|
|
self.description,
|
|
self.bank,
|
|
self.value,
|
|
self.category,
|
|
)
|
|
|
|
|
|
DbTransactions = list[DbTransaction]
|
|
|
|
CREATE_BACKUPS_TABLE = """
|
|
CREATE TABLE IF NOT EXISTS backups (
|
|
datetime TEXT NOT NULL,
|
|
file TEXT NOT NULL
|
|
)
|
|
"""
|
|
|
|
CREATE_BANKS_TABLE = """
|
|
CREATE TABLE IF NOT EXISTS banks (
|
|
name TEXT NOT NULL PRIMARY KEY,
|
|
bic TEXT,
|
|
nordigen_id TEXT,
|
|
nordigen_name TEXT,
|
|
requisition_id TEXT,
|
|
invert INTEGER
|
|
)
|
|
"""
|
|
|
|
|
|
@dataclass
|
|
class DbBank:
|
|
name: str
|
|
bic: str
|
|
nordigen_id: str
|
|
nordigen_name: str
|
|
requisition_id: str
|
|
invert: bool
|
|
|
|
def tuple(self):
|
|
return (
|
|
self.name,
|
|
self.bic,
|
|
self.nordigen_id,
|
|
self.nordigen_name,
|
|
self.requisition_id,
|
|
int(self.invert),
|
|
)
|
|
|
|
|
|
DbBanks = list[DbBank]
|
|
|
|
ADD_TRANSACTION = """
|
|
INSERT INTO transactions (date, description, bank, value, category) values (?,?,?,?,?)
|
|
"""
|
|
|
|
UPDATE_CATEGORY = """
|
|
UPDATE transactions
|
|
SET category = (?)
|
|
WHERE date = (?) AND description = (?) AND bank = (?) AND value = (?)
|
|
"""
|
|
|
|
DUPLICATED_TRANSACTIONS = """
|
|
SELECT COUNT(*), date, description, bank, value
|
|
FROM transactions
|
|
GROUP BY date, description, bank, value
|
|
HAVING COUNT(*) > 1
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SORTED_TRANSACTIONS = """
|
|
SELECT *
|
|
FROM transactions
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SELECT_TRANSACTIONS_BETWEEN_DATES = """
|
|
SELECT *
|
|
FROM transactions
|
|
WHERE date BETWEEN (?) AND (?)
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SELECT_TRANSACTIONS_BY_CATEGORY = """
|
|
SELECT *
|
|
FROM transactions
|
|
WHERE category IS (?)
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SELECT_TRANSACTIONS_BETWEEN_DATES_WITH_CATEGORY = """
|
|
SELECT *
|
|
FROM transactions
|
|
WHERE date BETWEEN (?) AND (?)
|
|
AND category IS (?)
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SELECT_TRANSACTION_BY_PERIOD = """
|
|
SELECT EXTRACT((?) FROM date) AS (?), date, description, bank, value
|
|
FROM transactions
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
SELECT_TRANSACTIONS_BETWEEN_DATES_WITHOUT_CATEGORIES = """
|
|
SELECT *
|
|
FROM transactions
|
|
WHERE date BETWEEN (?) AND (?)
|
|
AND category NOT IN {}
|
|
ORDER BY date ASC
|
|
"""
|
|
|
|
ADD_BANK = """
|
|
INSERT INTO banks (name, bic, nordigen_id, nordigen_name, requisition_id, invert) values (?,?,?,?,?,?)
|
|
"""
|
|
|
|
DELETE_BANK = """
|
|
DELETE FROM banks
|
|
WHERE name = (?)
|
|
"""
|
|
|
|
SELECT_BANK = """
|
|
SELECT *
|
|
FROM banks
|
|
WHERE {} = (?)
|
|
"""
|
|
|
|
SELECT_BANKS = """
|
|
SELECT *
|
|
FROM banks
|
|
"""
|