mafia.py

import sqlite3

conn = sqlite3.connect('mafia.db')

def init():
    try:
        db = conn.cursor()
        db.execute('create table if not exists user (id int primary key, name varchar(20))')
        db.close()
        db.commit()
        conn.close()
        print('用户数据不存在，正在创建...')
    except:
        print('用户数据已存在，加载用户数据...')

# def reg(name):
#     try:
#         db = conn.cursor()
#         db.execute('select * from user ')
#         db.execute('insert into user (id, name) values (2, ?)',(name,))
#         res = db.fetchall()
#         db.close()
#         conn.commit()
#         print(res)
#         db.close()
#     except:
#         return '系统错误'

def newGame():
    try:
        db = conn.cursor()
        db.execute('creat table game (player_id int primary key, player )')
        res = db.fetchall()
        db.close()
        conn.commit()
        print(res)
        db.close() 
        return '游戏创建成功，正在等待玩家加入...\n发送"加入"即可加入游戏。'
    except:
        return '当前有正在进行中的游戏，请结束游戏后在开新局。'