Compare commits

..

4 Commits

Author SHA1 Message Date
29ceb95404 Added db.py in which the PostgreSQLDatabase class were added for working with DB. 2023-05-27 16:17:29 +05:00
9924f0eebb Added main.py in which there was added config parser for retrieving token for Telegram bot, hotels.com API, and parameters for database.
Also added templates for bot commands and bot polling.
2023-05-27 16:15:16 +05:00
2de77f7c8a Revert "File templates added"
This reverts commit ff5952624deb4253ec7126e681ce869997a3e336.
2023-05-27 15:06:59 +05:00
ff5952624d File templates added 2023-05-27 15:06:35 +05:00
3 changed files with 70 additions and 0 deletions

2
.gitignore vendored
View File

@ -160,3 +160,5 @@ cython_debug/
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/
# Configuration file
config.ini

23
db.py Normal file
View File

@ -0,0 +1,23 @@
import psycopg2
class PostgreSQLDatabase:
def __init__(self, host, database, user, password):
self.__host = host
self.__database = database
self.__user = user
self.__password = password
def execute_query(self, query):
connection = psycopg2.connect(
host=self.__host,
database=self.__database,
user=self.__user,
password=self.__password
)
cursor = connection.cursor()
cursor.execute(query)
result = cursor.fetchall()
cursor.close()
connection.close()
return result

45
main.py Normal file
View File

@ -0,0 +1,45 @@
import telebot
import configparser
config = configparser.ConfigParser()
config.read()
TG_TOKEN = config.get('telebot', 'token')
API_TOKEN = config.get('hotels_api', 'token')
DB_HOST = config.get('database', 'host')
DB_PORT = config.get('database', 'port')
DB_DATABASE = config.get('database', 'database')
DB_USER = config.get('database', 'user')
DB_PASSWORD = config.get('database', 'password')
bot = telebot.TeleBot(token=TG_TOKEN)
@bot.message_handler(commands=['start', 'help'])
def handle_start(message):
pass # TODO добавить логику обработки команд start и help
@bot.message_handler(commands=['low'])
def handle_low(message):
pass # TODO добавить логику обработки команды low
@bot.message_handler(commands=['high'])
def handle_high(message):
pass # TODO добавить логику обработки команды high
@bot.message_handler(commands=['custom'])
def handle_custom(message):
pass # TODO добавить логику обработки команды custom
@bot.message_handler(commands=['history'])
def handle_history(message):
pass # TODO добавить логику обработки команды history
bot.polling()