Add SQLite and MongoDB implementations for YouTube video management with CRUD operations

Wcoder547 · Wcoder547 · commit 690a8dc5f107 · 2025-07-08T12:12:36.000+05:00
diff --git a/youtube_manager/youtube_videos.db b/youtube_manager/youtube_videos.db
diff --git a/youtube_manager/ytmanager_with_mongodb.py b/youtube_manager/ytmanager_with_mongodb.py
@@ -0,0 +1,59 @@
+from pymongo import MongoClient
+from bson import ObjectId
+
+client = MongoClient("mongodb+srv://CHAI:CHAI@cluster0.lxl3fsq.mongodb.net/", tlsAllowInvalidCertificates=True)
+# Not a good idea to include id and password in code files
+#  tlsAllowInvalidCertificates=True - Not a good way to handle ssl
+
+print(client)
+db = client["ytmanager"]
+video_collection = db["videos"]
+
+# print(video_collection)
+
+def add_video(name, time):
+    video_collection.insert_one({"name": name, "time": time})
+
+def list_videos():
+    for video in video_collection.find():
+        print(f"ID: {video['_id']}, Name: {video['name']} and Time: {video['time']}")
+
+def update_video(video_id, new_name, new_time):
+    video_collection.update_one({'_id': ObjectId(video_id)}, {"$set": {"name": new_name, "time": new_time}})
+
+def delete_video(video_id):
+    video_collection.delete_one({"_id": video_id})
+    # TODO: debug this video_id problem
+
+
+def main():
+    while True:
+        print("\n Youtube manager App")
+        print("1. List all videos")
+        print("2. Add a new videos")
+        print("3. Update a videos")
+        print("4. Delete a videos")
+        print("5. Exit the app")
+        choice = input("Enter your choice: ")
+
+        if choice == '1':
+            list_videos()
+        elif choice == '2':
+            name = input("Enter the video name: ")
+            time = input("Enter the video time: ")
+            add_video(name, time)
+        elif choice == '3':
+            video_id = input("Enter the video id to update: ")
+            name = input("Enter the updated video name: ")
+            time = input("Enter the updated video time: ")
+            update_video(video_id, name, time)
+        elif choice == '4':
+            video_id = input("Enter the video id to update: ")
+            delete_video(video_id, name, time)
+        elif choice == '5':
+            break
+        else:
+            print("Invalid choice")
+
+if __name__ == "__main__":
+    main()
diff --git a/youtube_manager/ytmanager_with_sqllite.py b/youtube_manager/ytmanager_with_sqllite.py
@@ -0,0 +1,64 @@
+import sqlite3
+
+conn = sqlite3.connect('youtube_videos.db')
+
+cursor = conn.cursor()
+
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS videos (
+               id INTEGER PRIMARY KEY,
+               name TEXT NOT NULL,
+               time TEXT NOT NULL
+    )
+''')
+
+def list_videos():
+    cursor.execute("SELECT * FROM videos")
+    for row in cursor.fetchall():
+        print(row)
+
+def add_video(name, time):
+    cursor.execute("INSERT INTO videos (name, time) VALUES (?, ?)", (name, time))
+    conn.commit()
+
+def update_video(video_id, new_name, new_time):
+    cursor.execute("UPDATE videos SET name = ?, time = ? WHERE id = ?", (new_name, new_time, video_id))
+    conn.commit()
+
+def delete_video(video_id):
+    cursor.execute("DELETE FROM videos where id = ?", (video_id,))
+    conn.commit()
+
+def main():
+    while True:
+        print("\n Youtube manager app with DB")
+        print("1. List Videos")
+        print("2. Add Videos")
+        print("3. Update Videos")
+        print("4. Delete Videos")
+        print("5. exit app")
+        choice = input("Enter your choice: ")
+
+        if choice == '1':
+            list_videos()
+        elif choice == '2':
+            name = input("Enter the video name: ")
+            time = input("Enter the video time: ")
+            add_video(name, time)
+        elif choice == '3':
+            video_id = input("Enter video ID to update: ")
+            name = input("Enter the video name: ")
+            time = input("Enter the video time: ")
+            update_video(video_id, name, time)
+        elif choice == '4':
+            video_id = input("Enter video ID to delete: ")
+            delete_video(video_id)
+        elif choice == '5':
+            break
+        else:
+            print("Invalid Choice ")
+
+    conn.close()
+
+if __name__ == "__main__":
+    main()
