pass-gen/app/repositories/sqlite.py
2021-08-03 14:28:58 +02:00

22 lines
629 B
Python

import app.data.sqlite as sqlite
from app.data.type import DBConnector
class PasswordRepository:
def __init__(self, db: DBConnector) -> None:
self.db = db
def save(self, password: str) -> None:
try:
self.db.execute("INSERT INTO passwords VALUES (:password)", {"password": password})
self.db.commit()
except Exception as e:
print(e)
def list_all(self) -> list[str]:
return [row[0] for row in self.db.execute("SELECT * FROM passwords").fetchall()]
def get_instance() -> PasswordRepository:
db = sqlite.DB()
return PasswordRepository(db)