44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from .template import Repo
|
|
from dataclasses import dataclass
|
|
from exceptions import not_found
|
|
|
|
|
|
@dataclass
|
|
class Prefix:
|
|
name: str
|
|
color: str = ""
|
|
weight: int = 0
|
|
|
|
|
|
class PrefixRepo(Repo):
|
|
def get_all(self) -> list[Prefix]:
|
|
self.cur.execute("SELECT * FROM prefixes ORDER BY weight, name")
|
|
results = self.cur.fetchall()
|
|
if not results:
|
|
return []
|
|
return [Prefix(*r) for r in results]
|
|
|
|
def get_one(self, prefix_name: str):
|
|
self.cur.execute("SELECT * FROM prefixes WHERE name = %s", (prefix_name,))
|
|
result = self.cur.fetchone()
|
|
if not result:
|
|
raise not_found
|
|
return Prefix(*result)
|
|
|
|
def add_one(self, prefix: Prefix) -> str:
|
|
self.cur.execute("REPLACE INTO prefixes VALUES (%s, %s, %s)", (prefix.name, prefix.color, prefix.weight))
|
|
self.conn.commit()
|
|
return "Prefix inserted successfully."
|
|
|
|
def add_list(self, prefix_list: list[Prefix]):
|
|
for prefix in prefix_list:
|
|
self.cur.execute("REPLACE INTO prefixes VALUES (%s, %s, %s)", (prefix.name, prefix.color, prefix.weight))
|
|
self.conn.commit()
|
|
return "Prefixes inserted successfully."
|
|
|
|
|
|
def del_one(self, prefix_name: str) -> str:
|
|
self.cur.execute("DELETE FROM prefixes WHERE name = %s", (prefix_name,))
|
|
self.conn.commit()
|
|
return "Prefix deleted successfully."
|