Create database.py

database.py

@@ -0,0 +1,43 @@
+import sqlite3
+import os
+
+DATABASE = 'database.db'
+
+def create_table():
+    with sqlite3.connect(DATABASE) as db:
+        cursor = db.cursor()
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS news (
+                id INTEGER PRIMARY KEY,
+                title TEXT NOT NULL,
+                source TEXT NOT NULL,
+                published TEXT NOT NULL,
+                url TEXT NOT NULL,
+                summary TEXT NOT NULL,
+                content TEXT NOT NULL
+            )
+        ''')
+        db.commit()
+
+def insert_news(title, source, published, url, summary, content):
+    with sqlite3.connect(DATABASE) as db:
+        cursor = db.cursor()
+        cursor.execute('''
+            INSERT INTO news (title, source, published, url, summary, content)
+            VALUES (?,?,?,?,?,?)
+        ''', (title, source, published, url, summary, content))
+        db.commit()
+
+def get_all_news():
+    with sqlite3.connect(DATABASE) as db:
+        cursor = db.cursor()
+        cursor.execute('SELECT * FROM news')
+        return cursor.fetchall()
+
+def get_news_by_id(id):
+    with sqlite3.connect(DATABASE) as db:
+        cursor = db.cursor()
+        cursor.execute('SELECT * FROM news WHERE id =?', (id,))
+        return cursor.fetchone()
+
+create_table()