S7eAHnHT

1import pymysql
2pymysql.install_as_MySQLdb()
3
4class DB:
5    def __init__(self):
6        config = {
7            'host': 'localhost',
8            'port': 3306,
9            'user': 'root',
10            'passwd': 'CloudF&&
11            
12            Zanilia',
13            'db': 'test',
14            'charset': 'utf8',
15        }
16        self.conn = pymysql.connect(**config)
17
18        # 创建没加前缀的数据表
19        sql = '''CREATE TABLE IF NOT EXISTS no_prefix_remind_list(
20        tid VARCHAR(20) PRIMARY KEY)
21        '''
22        cursor = self.conn.cursor()
23        cursor.execute(sql)
24
25        # 创建坟贴数据表
26        grave_remind_tid_sql = '''CREATE TABLE IF NOT EXISTS grave_remind_tid_list(
27        id INTEGER PRIMARY KEY AUTO_INCREMENT,
28        tid VARCHAR(20),
29        last_replier VARCHAR(15) ,
30        INDEX(tid)
31        )
32        '''
33
34        cursor.execute(grave_remind_tid_sql)
35        cursor.close()
36
37    def is_exit_in_prefix_list(self, tid):
38        cursor = self.conn.cursor()
39        sql = '''SELECT * FROM no_prefix_remind_list WHERE tid='%s' ''' % tid
40        num = cursor.execute(sql)
41        cursor.close()
42        return bool(num)
43
44    def insert_in_prefix_list(self, tid):
45        cursor = self.conn.cursor()
46        sql = '''INSERT INTO no_prefix_remind_list (tid) VALUES ('%s')
47        ''' % tid
48        cursor.execute(sql)
49        self.conn.commit()
50        cursor.close()
51
52    def is_exit_in_grave_list(self, tid, last_replier):
53        sql = '''SELECT * FROM grave_remind_tid_list WHERE tid='%s' AND last_replier='%s' ''' % (tid, last_replier)
54        cursor = self.conn.cursor()
55        result = bool(cursor.execute(sql))
56        cursor.close()
57        return result
58
59    def insert_in_grave_list(self, tid, last_replier):
60        cursor = self.conn.cursor()
61        sql = '''INSERT INTO grave_remind_tid_list (tid,last_replier) VALUES ('%s','%s')''' % (tid, last_replier)
62        cursor.execute(sql)
63        self.conn.commit()
64        cursor.close()
65
66db = DB()
67if __name__ == '__main__':
68    for i in range(10):
69        print(db.is_exit_in_grave_list('5509495784', '倾妃晗'))