Archived
1
0
Fork 0
This repository has been archived on 2024-04-26. You can view files and clone it, but cannot push or open issues or pull requests.
akari-bot/modules/minecraft_news/database.py
2021-06-01 22:38:03 +08:00

43 lines
1.1 KiB
Python

import base64
import os
import re
import sqlite3
import traceback
dbpath = os.path.abspath('./modules/minecraft_news/save.db')
class MD:
def __init__(self):
if not os.path.exists(dbpath):
self.initialize()
self.conn = sqlite3.connect(dbpath)
self.c = self.conn.cursor()
def initialize(self):
a = open(dbpath, 'w')
a.close()
self.conn = sqlite3.connect(dbpath)
self.c = self.conn.cursor()
self.c.execute('''CREATE TABLE mc_news
(TITLE TEXT PRIMARY KEY NOT NULL,
LINK TEXT,
PDESC TEXT,
IMAGE TEXT,
PDATE TEXT);''')
self.c.close()
def add_news(self, title, link, desc, image, date):
self.c.execute(f"INSERT INTO mc_news (TITLE, LINK, PDESC, IMAGE, PDATE) VALUES (?, ?, ?, ?, ?)", (title, link, desc, image, date))
self.conn.commit()
return
def check_exist(self, title):
a = self.c.execute(f"SELECT * FROM mc_news WHERE TITLE=?", (title,)).fetchone()
if a:
return True
else:
return False