new rooms schema

createRooms.py → stablediffusion-infinity/createRooms.py

@@ -9,12 +9,20 @@ import json
 app = FastAPI()
 LIVEBLOCKS_SECRET = os.environ.get("LIVEBLOCKS_SECRET")
 
+DB_PATH = Path("rooms.db")
+if not DB_PATH.exists():
+    print("Creating database")
+    print("DB_PATH", DB_PATH)
+    db = sqlite3.connect(DB_PATH)
+    with open(Path("schema.sql"), "r") as f:
+        db.executescript(f.read())
+    db.commit()
+    db.close()
+
 
 def get_db():
     db = sqlite3.connect(Path("./rooms.db"), check_same_thread=False)
-    db.execute("CREATE TABLE IF NOT EXISTS rooms (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, room_id TEXT NOT NULL, users_count INTEGER NOT NULL DEFAULT 0)")
     print("Connected to database")
-    db.commit()
     db.row_factory = sqlite3.Row
     try:
         yield db
@@ -26,7 +34,7 @@ def get_db():
 
 app = FastAPI()
 
-rooms = ["sd-multiplayer-room-" + str(i) for i in range(0, 20)]
+rooms = ["room-" + str(i) for i in range(0, 41)]
 
 
 @app.get("/")
@@ -109,4 +117,4 @@ async def sync_rooms(db: sqlite3.Connection = Depends(get_db)):
 
 
 if __name__ == "__main__":
-    uvicorn.run("api:app", host="0.0.0.0", log_level="debug", reload=True)
+    uvicorn.run("createRooms:app", host="0.0.0.0", log_level="debug", reload=True)

stablediffusion-infinity/schema.sql

@@ -1,26 +1,47 @@
 PRAGMA foreign_keys=OFF;
 BEGIN TRANSACTION;
 CREATE TABLE rooms (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, room_id TEXT NOT NULL, users_count INTEGER NOT NULL DEFAULT 0);
-INSERT INTO rooms VALUES(1,'sd-multiplayer-room-0',0);
-INSERT INTO rooms VALUES(2,'sd-multiplayer-room-1',0);
-INSERT INTO rooms VALUES(3,'sd-multiplayer-room-2',0);
-INSERT INTO rooms VALUES(4,'sd-multiplayer-room-3',0);
-INSERT INTO rooms VALUES(5,'sd-multiplayer-room-4',0);
-INSERT INTO rooms VALUES(6,'sd-multiplayer-room-5',0);
-INSERT INTO rooms VALUES(7,'sd-multiplayer-room-6',0);
-INSERT INTO rooms VALUES(8,'sd-multiplayer-room-7',0);
-INSERT INTO rooms VALUES(9,'sd-multiplayer-room-8',0);
-INSERT INTO rooms VALUES(10,'sd-multiplayer-room-9',0);
-INSERT INTO rooms VALUES(11,'sd-multiplayer-room-10',0);
-INSERT INTO rooms VALUES(12,'sd-multiplayer-room-11',0);
-INSERT INTO rooms VALUES(13,'sd-multiplayer-room-12',0);
-INSERT INTO rooms VALUES(14,'sd-multiplayer-room-13',0);
-INSERT INTO rooms VALUES(15,'sd-multiplayer-room-14',0);
-INSERT INTO rooms VALUES(16,'sd-multiplayer-room-15',0);
-INSERT INTO rooms VALUES(17,'sd-multiplayer-room-16',0);
-INSERT INTO rooms VALUES(18,'sd-multiplayer-room-17',0);
-INSERT INTO rooms VALUES(19,'sd-multiplayer-room-18',0);
-INSERT INTO rooms VALUES(20,'sd-multiplayer-room-19',0);
+INSERT INTO rooms VALUES(1,'room-0',0);
+INSERT INTO rooms VALUES(2,'room-1',0);
+INSERT INTO rooms VALUES(3,'room-2',0);
+INSERT INTO rooms VALUES(4,'room-3',0);
+INSERT INTO rooms VALUES(5,'room-4',0);
+INSERT INTO rooms VALUES(6,'room-5',0);
+INSERT INTO rooms VALUES(7,'room-6',0);
+INSERT INTO rooms VALUES(8,'room-7',0);
+INSERT INTO rooms VALUES(9,'room-8',0);
+INSERT INTO rooms VALUES(10,'room-9',0);
+INSERT INTO rooms VALUES(11,'room-10',0);
+INSERT INTO rooms VALUES(12,'room-11',0);
+INSERT INTO rooms VALUES(13,'room-12',0);
+INSERT INTO rooms VALUES(14,'room-13',0);
+INSERT INTO rooms VALUES(15,'room-14',0);
+INSERT INTO rooms VALUES(16,'room-15',0);
+INSERT INTO rooms VALUES(17,'room-16',0);
+INSERT INTO rooms VALUES(18,'room-17',0);
+INSERT INTO rooms VALUES(19,'room-18',0);
+INSERT INTO rooms VALUES(20,'room-19',0);
+INSERT INTO rooms VALUES(21,'room-20',0);
+INSERT INTO rooms VALUES(22,'room-21',0);
+INSERT INTO rooms VALUES(23,'room-22',0);
+INSERT INTO rooms VALUES(24,'room-23',0);
+INSERT INTO rooms VALUES(25,'room-24',0);
+INSERT INTO rooms VALUES(26,'room-25',0);
+INSERT INTO rooms VALUES(27,'room-26',0);
+INSERT INTO rooms VALUES(28,'room-27',0);
+INSERT INTO rooms VALUES(29,'room-28',0);
+INSERT INTO rooms VALUES(30,'room-29',0);
+INSERT INTO rooms VALUES(31,'room-30',0);
+INSERT INTO rooms VALUES(32,'room-31',0);
+INSERT INTO rooms VALUES(33,'room-32',0);
+INSERT INTO rooms VALUES(34,'room-33',0);
+INSERT INTO rooms VALUES(35,'room-34',0);
+INSERT INTO rooms VALUES(36,'room-35',0);
+INSERT INTO rooms VALUES(37,'room-36',0);
+INSERT INTO rooms VALUES(38,'room-37',0);
+INSERT INTO rooms VALUES(39,'room-38',0);
+INSERT INTO rooms VALUES(40,'room-39',0);
+INSERT INTO rooms VALUES(41,'room-40',0);
 DELETE FROM sqlite_sequence;
-INSERT INTO sqlite_sequence VALUES('rooms',20);
+INSERT INTO sqlite_sequence VALUES('rooms',41);
 COMMIT;