app.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
#  Matebot
#  
#  Copyleft 2012-2021 Iuri Guilherme <https://github.com/iuriguilherme>,
#     Matehackers <https://github.com/matehackers>
#  
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#  
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#  
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#  
## Do jeito mais difícil para o mais fácil:
##  1) ./start.py
##  2) python3 start.py
##  3) pipenv run start.py aiogram matebot
##  4) pipenv run matebot
##  5) Ler o README.md ;)

### Logging
import logging
logging.basicConfig(level=logging.INFO)
# ~ logging.basicConfig(level=logging.DEBUG)

import subprocess, sys

from matebot.aio_matebot import run as aiogram_bot

try:
  from matebot.ptb_matebot import app
  flask_bot = app
except Exception as e:
  logging.debug(repr(e))

try:
  from matebot.tp_matebot import bot as telepot_bot
except Exception as e:
  logging.debug(repr(e))

if __name__ == "__main__":
  mode = 'aiogram'
  bot = 'matebot'
  port = 5000
  ## TODO fazer validação de verdade
  if len(sys.argv) > 1:
    mode = sys.argv[1]
    print(u"Modo de operação: %s" % (mode))
    if len(sys.argv) > 2:
      bot = sys.argv[2]
      print(
        u"Usando token do bot \"{}\" do arquivo de configuração.\
          ".format(bot)
      )
      if len(sys.argv) > 3:
        port = sys.argv[3]
        print(u"Alterando porta para {}".format(port))
    else:
      print(u"Nome do bot não informado, {} presumido".format(bot))
  else:
    print(u"Modo de operação não informado, {} presumido.".format(mode))
    print(u"Nome do bot não informado, {} presumido".format(bot))
  if mode == 'aiogram':
    aiogram_bot(bot)
  elif mode == 'flask':
    flask_bot.run(port=port)
  elif mode == 'telepot':
    telepot_bot(mode, bot)
  else:
    print(u"Não entendi nada, não consegui iniciar. Leia o manual por favor.")