38 lines
968 B
Python
38 lines
968 B
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from peewee import *
|
|
|
|
db = SqliteDatabase('tgmsbot.db', pragmas={
|
|
'journal_mode': 'wal',
|
|
'cache_size': -32 * 1000})
|
|
|
|
class Player(Model):
|
|
user_id = IntegerField(unique=True, primary_key=True)
|
|
mines = IntegerField()
|
|
death = IntegerField()
|
|
wins = IntegerField()
|
|
restricted_until = IntegerField()
|
|
immunity_cards = IntegerField()
|
|
class Meta:
|
|
database = db
|
|
def save(self, *args, **kwargs):
|
|
super().save(*args, **kwargs)
|
|
db.close()
|
|
@staticmethod
|
|
def db_close():
|
|
db.close()
|
|
|
|
db.connect()
|
|
db.create_tables([Player])
|
|
db.close()
|
|
|
|
def get_player(user_id):
|
|
db.connect()
|
|
player = Player.get_or_none(Player.user_id == user_id)
|
|
if player is None:
|
|
player = Player.create(user_id=user_id, mines=0, death=0, wins=0,
|
|
restricted_until=0, immunity_cards=0)
|
|
return player
|
|
else:
|
|
return player
|