126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
"""Example showing how to run more bots with davtelepot.
|
|
|
|
In this example, one bot will ask for updates with long polling and another
|
|
one will receive them via webhook.
|
|
"""
|
|
|
|
# Standard library modules
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
# Third party modules
|
|
try:
|
|
from davtelepot.bot import Bot
|
|
except ImportError:
|
|
logging.error(
|
|
"Please install davtelepot library.\n"
|
|
"Using a python virtual environment is advised.\n\n"
|
|
"```bash\n"
|
|
"pip -m venv env\n"
|
|
"env/bin/pip install davtelepot\n"
|
|
"env/bin/python davtelepot/examples/a_simple_bot.py"
|
|
"```"
|
|
)
|
|
sys.exit(1)
|
|
|
|
# Project modules
|
|
from a_simple_bot import initialize_bot
|
|
|
|
# Get path of current script
|
|
path = os.path.dirname(__file__)
|
|
|
|
|
|
def _main():
|
|
# Import or prompt user for bot token
|
|
try:
|
|
from secrets import webhook_bot_token
|
|
except ImportError:
|
|
webhook_bot_token = input("Enter bot token:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(f'webhook_bot_token = "{webhook_bot_token}"\n')
|
|
try:
|
|
from secrets import hostname
|
|
except ImportError:
|
|
hostname = input("Enter host name:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(f'hostname = "{hostname}"\n')
|
|
try:
|
|
from secrets import certificate
|
|
except ImportError:
|
|
certificate = input("Enter ssl certificate:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(f'certificate = "{certificate}"\n')
|
|
try:
|
|
from secrets import local_host
|
|
except ImportError:
|
|
local_host = input("Enter local host:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(f'local_host = "{local_host}"\n')
|
|
try:
|
|
from secrets import port
|
|
except ImportError:
|
|
port = input("Enter local port:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(f'port = "{port}"\n')
|
|
try:
|
|
from secrets import long_polling_bot_token
|
|
except ImportError:
|
|
long_polling_bot_token = input("Enter bot token:\t\t")
|
|
with open(
|
|
f'{path}/secrets.py',
|
|
'a' # Append to file, create it if it does not exist
|
|
) as secrets_file:
|
|
secrets_file.write(
|
|
f'long_polling_bot_token = "{long_polling_bot_token}"\n'
|
|
)
|
|
# Set logging preferences
|
|
log_formatter = logging.Formatter(
|
|
"%(asctime)s [%(module)-15s %(levelname)-8s] %(message)s",
|
|
style='%'
|
|
)
|
|
root_logger = logging.getLogger()
|
|
root_logger.setLevel(logging.DEBUG)
|
|
consoleHandler = logging.StreamHandler()
|
|
consoleHandler.setFormatter(log_formatter)
|
|
consoleHandler.setLevel(logging.DEBUG)
|
|
root_logger.addHandler(consoleHandler)
|
|
|
|
# Instantiate, initialize and make bots run.
|
|
webhook_bot = Bot(
|
|
token=webhook_bot_token,
|
|
database_url=f"{path}/webhook_bot.db",
|
|
hostname=hostname,
|
|
certificate=certificate
|
|
)
|
|
initialize_bot(webhook_bot)
|
|
long_polling_bot = Bot(
|
|
token=long_polling_bot_token,
|
|
database_url=f"{path}/long_polling_bot.db",
|
|
)
|
|
initialize_bot(long_polling_bot)
|
|
logging.info("Send a KeyboardInterrupt (ctrl+C) to stop bots.")
|
|
Bot.run(
|
|
local_host=local_host,
|
|
port=port
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
_main()
|