def create_tables():
database.connect()
try:
User.create_table()
except: # OperationalError
pass # database already exists. skip creating tables.
database.close()
def restart(self):
try:
self.server.close()
except Exception as e:
logging.error('BAZARR Cannot stop Waitress: ' + repr(e))
else:
database.close()
try:
restart_file = io.open(os.path.join(args.config_dir, "bazarr.restart"), "w", encoding='UTF-8')
except Exception as e:
logging.error('BAZARR Cannot create bazarr.restart file: ' + repr(e))
else:
logging.info('Bazarr is being restarted...')
restart_file.write(str(''))
restart_file.close()
os._exit(0)
def adduser(_=None):
create_tables()
username = raw_input('Username:')
user = None
try:
user = User.get(name=username)
except User.DoesNotExist:
pass # ok
if user:
sys.exit("The username has been already used.")
password = getpass.getpass()
user = User(name=username, password=password)
user.save()
print("The user was created successfully.")
database.close()
async def finish():
if not MySQL.is_closed():
MySQL.close()
import os
import dotenv
from init.bot import Bot
from database.database import build, close
import traceback
dotenv.load_dotenv()
TOKEN = os.getenv("TOKEN")
bot = Bot()
for file in os.listdir("cogs"):
if file.endswith(".py"):
name = file[:-3]
bot.load_extension(f"cogs.{name}")
if build() == False:
exit(1)
try:
bot.run(TOKEN)
except:
traceback.print_exc()
close()
def shutdown():
if not connection.is_closed():
connection.close()
#encoding:utf8
#初始时用来创建所有的表的脚本
from database import database
#创建表user
database.query("""create table if not exists user
(uname varchar(15) not null,
email varchar(40) not null,
rname varchar(24) not null,
pwd varchar(32) not null,
permission enum('0', '1', '2', '3', '4') not null,
primary key(uname))""")
database.query("""create table if not exists session (
sessionid char(27),
sessionkey char(20),
uname varchar(15),
lastactive timestamp,
primary key(sessionid),
foreign key(uname) references user(uname)
)""")
database.query("""create table if not exists physical(
id int(10) auto_increment,
type_id int(10),
value float,
collect_time timestamp,
primary key(id)
)""")
database.close()
