frisbeer-bot/main.py at master · Uular/frisbeer-bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
import json
import logging
import sys

from sqlalchemy.orm import sessionmaker
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, ReplyKeyboardMarkup
from telegram.ext import Updater, CommandHandler, CallbackQueryHandler
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from player import Player

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.DEBUG,
                    filename="log.txt")


def start(bot, update):
    update.message.reply_text('Lets play frisbeer!\n Start with /rank <nick>')


def game(bot, update):
    keyboard = [
        [InlineKeyboardButton("New game", callback_data=json.dumps({"action": "create"})),
         InlineKeyboardButton("List pending games", callback_data=json.dumps({"action": "list"}))]
                ]
    reply_markup = InlineKeyboardMarkup(keyboard)

    update.message.reply_text('Please choose:', reply_markup=reply_markup)


def button(bot, update):
    query = update.callback_query
    data = json.loads(query.data)
    keyboard = [
        [InlineKeyboardButton("#norsu, 1.6. klo 18:15 5/6", callback_data=json.dumps({"action": "inspect_game", "id": 1}))],
        [InlineKeyboardButton("#hippo, 5.5. 0/6", callback_data=json.dumps({"action": "inspect_game", "id": 2}))]
    ]

    reply_markup = InlineKeyboardMarkup(keyboard)

    query.message.edit_text('Please choose:', reply_markup=reply_markup)


def rank(bot, update):
    usage = "Usage: /rank <nick> \nor register your frisbeer nick with /register <nick>"
    reply = update.message.reply_text
    logging.info("Rank query")
    logging.debug(update.message.text)
    try:
        nick = update.message.text.split("/rank ", 1)[1]
    except IndexError:
        # No nick provided by user
        telegram_username = update.message.from_user.username
        if not telegram_username or not len(telegram_username):
            reply(usage)
            return
        user = session.query(User).filter(User.telegram_username == update.message.from_user.username).first()
        if not user:
            reply(usage)
            return
        nick = user.frisbeer_nick
    else:
        # Provided nick may be a Telegram username
        if nick.startswith('@') and len(nick) > 1:
            user = session.query(User).filter(User.telegram_username == nick[1:]).first()
            if user:
                nick = user.frisbeer_nick

    player = Player.by_nick(nick)
    reply(str(player))


def register(bot, update):
    logging.info("Registering nick")
    logging.debug(update.message.text)
    logging.debug(update.message.from_user.username)
    reply = update.message.reply_text
    telegram_username = update.message.from_user.username

    if not telegram_username:
        reply("First set up a Telegram username in settings")
        return

    try:
        frisbeer_nick = update.message.text.split("register ", 1)[1]
    except IndexError:
        reply("Usage: /register <frisbeer nick>")
        return

    user = session.query(User).filter(User.telegram_username == telegram_username).first()
    if user is None:
        user = User(telegram_username=telegram_username, frisbeer_nick=frisbeer_nick)
        session.add(user)
        session.commit()
        reply("Paired {} with your username".format(frisbeer_nick))
    else:
        user.frisbeer_nick = frisbeer_nick
        session.commit()
        reply("Updated nick to {}".format(frisbeer_nick))


engine = create_engine('sqlite:///db.sqlite3', echo=True)
Base = declarative_base()


class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    frisbeer_nick = Column(String)
    telegram_username = Column(String, unique=True)

    def __repr__(self):
        return "{} - {}".format(self.telegram_nick, self.frisbeer_nick)


Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()

updater = Updater(sys.argv[1])

updater.dispatcher.add_handler(CommandHandler('start', start))
updater.dispatcher.add_handler(CommandHandler('game', game))
updater.dispatcher.add_handler(CommandHandler('rank', rank))
updater.dispatcher.add_handler(CommandHandler('register', register))
updater.dispatcher.add_handler(CallbackQueryHandler(button))

updater.start_polling()
updater.idle()