working on project

models/city.py

@@ -1,7 +1,7 @@
 import sqlite3
 
 class City:
-    def __init__(self, connection : sqlite3.Connection, id, name):
+    def __init__(self, connection : sqlite3.Connection, id = None, name = None):
         self.connection = connection
         self.id = id
         self.name = name
@@ -55,11 +55,12 @@ class City:
     
     def delete(self):
         cur = self.connection.cursor()
-        cur.execute("""DELETE FROM logs WHERE id = ?""", (self.id))
+        cur.execute("""DELETE FROM cities WHERE id = ?""", (self.id))
         self.connection.commit()
         cur.close()
         return True
     
     def _set_row_values(self, row):
         self.id = row[0]
-        self.name = row[1]
+        self.name = row[1]
+        return self