commit

app/managers.py

@@ -0,0 +1,65 @@
+import sqlite3
+
+from models import Actor
+
+
+class ActorManager:
+ def __init__(self) -> None:
+ self._connection = sqlite3.connect("cinema.db")
+ self.create_table()
+ self.table_name = "actors"
+
+ def create_table(self) -> None:
+ self._connection.execute(
+ """
+ CREATE TABLE IF NOT EXISTS actors (
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
+ first_name TEXT NOT NULL,
+ last_name TEXT NOT NULL
+ )
+ """
+ )
+ self._connection.commit()
+
+ def create(self, first_name: str, last_name: str) -> None:
+ self._connection.execute(
+ f"""
+ INSERT INTO {self.table_name}
+ (first_name, last_name) VALUES (?, ?)
+ """,
+ (first_name, last_name)
+ )
+ self._connection.commit()
+
+ def all(self) -> list:
+ cinema_cursor = self._connection.execute(
+ "SELECT * FROM actors"
+ )
+ return [
+ Actor(*row) for row in cinema_cursor
+ ]
+
+ def update(
+ self,
+ id_: int,
+ new_first_name: str,
+ new_last_name: str
+ ) -> None:
+ self._connection.execute(
+ f"""
+ UPDATE {self.table_name}
+ SET first_name = ?, last_name = ?
+ WHERE id = ?""",
+ (new_first_name, new_last_name, id_)
+ )
+ self._connection.commit()
+
+ def delete(self, id_delete: int) -> None:
+ self._connection.execute(
+ f"""
+ DELETE FROM {self.table_name}
+ WHERE id = ?
+ """,
+ (id_delete,)
+ )
+ self._connection.commit()

app/models.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class Actor:
+ id: int
+ first_name: str
+ last_name: str