Added testing and premium accounts and commands
This commit is contained in:
Joan Cano
@@ -39,97 +39,115 @@ async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
|
||||
Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`\n
|
||||
Lista los productos con `/list` o obtén la información de uno en concreto con `/list nombre del producto`\n
|
||||
Borra un producto con `/remove nombre del producto`"""
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
|
||||
else:
|
||||
message = """Activa tu periodo de prueba de 7 días con `/test` o contacta con @jocarduck para más información."""
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
|
||||
|
||||
async def add_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
telegram_user_id = helpers.get_telegram_user_id(update)
|
||||
if walladb.is_user_valid(telegram_user_id):
|
||||
number_of_products = walladb.count_user_products(telegram_user_id)
|
||||
message = """Tienes que pasar el número correcto de parámetros: `/add producto;precio_mínimo;precio_máximo,excluir_título(opcional, separado por comas);excluir_descripción_y_título(opciona, separado por comas);latitud(opcional);longitud(opcional),distancia(opcional)`\n
|
||||
Ejemplo: `/add placa base itx;0;150`\n
|
||||
Ejemplo 2: `/add cpu;10;30;;intel,core 2 duo,celeron;;;100`\n
|
||||
Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`"""
|
||||
|
||||
args = update.message.text.split("/add ")
|
||||
if len(args) == 1:
|
||||
pass
|
||||
elif len(args[1].split(";")) > 2:
|
||||
product = dict()
|
||||
product['telegram_user_id'] = telegram_user_id
|
||||
args = args[1].split(";")
|
||||
product['product_name'], product['min_price'], product['max_price'] = args[0:3]
|
||||
if len(args) > 3 and args[3]:
|
||||
product['title_exclude'] = args[3]
|
||||
if len(args) > 4 and args[4]:
|
||||
product['title_description_exclude'] = args[4]
|
||||
if len(args) > 5 and args[5]:
|
||||
product['latitude'] = args[5]
|
||||
if len(args) > 6 and args[6]:
|
||||
product['longitude'] = args[6]
|
||||
if len(args) > 7 and args[7]:
|
||||
product['distance'] = args[7]
|
||||
|
||||
logging.info(f'Adding: {product}')
|
||||
if not walladb.get_product(product):
|
||||
walladb.add_product(product)
|
||||
p = threading.Thread(target=Worker.run, args=(walladb.get_product(product), ))
|
||||
p.start()
|
||||
message = f"Añadido {walladb.get_product(product)['product_name']} a seguimiento"
|
||||
else:
|
||||
message = f"{walladb.get_product(product)['product_name']} ya está en seguimiento!"
|
||||
valid = False
|
||||
if walladb.is_user_testing(telegram_user_id):
|
||||
valid = True
|
||||
if number_of_products > 5:
|
||||
message = "Tienes más de 5 productos en seguimiento. Con premium puedes tener hasta 20."
|
||||
valid = False
|
||||
elif walladb.is_user_premium(telegram_user_id):
|
||||
valid = True
|
||||
if number_of_products > 20:
|
||||
message = "Tienes más de 20 productos en seguimiento. Borra algunos para añadir más."
|
||||
valid = False
|
||||
if valid:
|
||||
args = update.message.text.split("/add ")
|
||||
if len(args) == 1:
|
||||
pass
|
||||
elif len(args[1].split(";")) > 2:
|
||||
product = dict()
|
||||
product['telegram_user_id'] = telegram_user_id
|
||||
args = args[1].split(";")
|
||||
product['product_name'], product['min_price'], product['max_price'] = args[0:3]
|
||||
if len(args) > 3 and args[3]:
|
||||
product['title_exclude'] = args[3]
|
||||
if len(args) > 4 and args[4]:
|
||||
product['title_description_exclude'] = args[4]
|
||||
if len(args) > 5 and args[5]:
|
||||
product['latitude'] = args[5]
|
||||
if len(args) > 6 and args[6]:
|
||||
product['longitude'] = args[6]
|
||||
if len(args) > 7 and args[7]:
|
||||
product['distance'] = args[7]
|
||||
|
||||
logging.info(f'Adding: {product}')
|
||||
if not walladb.get_product(product):
|
||||
walladb.add_product(product)
|
||||
p = threading.Thread(target=Worker.run, args=(walladb.get_product(product), ))
|
||||
p.start()
|
||||
message = f"Añadido {walladb.get_product(product)['product_name']} a seguimiento."
|
||||
else:
|
||||
message = f"¡{walladb.get_product(product)['product_name']} ya está en seguimiento!"
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
|
||||
|
||||
async def remove_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
telegram_user_id = helpers.get_telegram_user_id(update)
|
||||
if walladb.is_user_valid(telegram_user_id):
|
||||
product_to_remove = update.message.text[len('/remove '):]
|
||||
message = f"{product_to_remove} no está en seguimiento!"
|
||||
if walladb.remove_product({'product_name' : product_to_remove, \
|
||||
'telegram_user_id' : telegram_user_id}):
|
||||
message = f"{product_to_remove} borrado!"
|
||||
await update.message.reply_text(message)
|
||||
if walladb.is_user_testing(telegram_user_id) or walladb.is_user_premium(telegram_user_id):
|
||||
product_to_remove = update.message.text[len('/remove '):]
|
||||
message = f"¡{product_to_remove} no está en seguimiento!"
|
||||
if walladb.remove_product({'product_name' : product_to_remove, \
|
||||
'telegram_user_id' : telegram_user_id}):
|
||||
message = f"¡{product_to_remove} borrado de la lista de seguimiento!"
|
||||
await update.message.reply_text(message)
|
||||
|
||||
async def list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
telegram_user_id = helpers.get_telegram_user_id(update)
|
||||
if walladb.is_user_valid(telegram_user_id):
|
||||
products = walladb.get_products_from_user(telegram_user_id)
|
||||
if walladb.is_user_testing(telegram_user_id) or walladb.is_user_premium(telegram_user_id):
|
||||
products = walladb.get_products_from_user(telegram_user_id)
|
||||
|
||||
args = update.message.text.split("/list ")
|
||||
found = False
|
||||
if len(args) > 1:
|
||||
product = walladb.get_product({'product_name':args[1],'telegram_user_id':telegram_user_id})
|
||||
if product:
|
||||
table = prettytable.PrettyTable(['Campo', 'Valor'])
|
||||
table.align['Campo'] = 'l'
|
||||
table.align['Valor'] = 'r'
|
||||
for key in product:
|
||||
table.add_row([key, product[key]])
|
||||
found = True
|
||||
if not found:
|
||||
table = prettytable.PrettyTable(['Producto', 'Mín', 'Máx'])
|
||||
table.align['Producto'] = 'l'
|
||||
table.align['Mín'] = 'r'
|
||||
table.align['Máx'] = 'r'
|
||||
for product in products:
|
||||
table.add_row([helpers.telegram_escape_characters(product['product_name']), f"{helpers.telegram_escape_characters(product['min_price'])}€", f"{helpers.telegram_escape_characters(product['max_price'])}€"])
|
||||
await update.message.reply_markdown_v2(f'```{(table)}```')
|
||||
args = update.message.text.split("/list ")
|
||||
found = False
|
||||
if len(args) > 1:
|
||||
product = walladb.get_product({'product_name':args[1],'telegram_user_id':telegram_user_id})
|
||||
if product:
|
||||
table = prettytable.PrettyTable(['Campo', 'Valor'])
|
||||
table.align['Campo'] = 'l'
|
||||
table.align['Valor'] = 'r'
|
||||
for key in product:
|
||||
table.add_row([key, product[key]])
|
||||
found = True
|
||||
if not found:
|
||||
table = prettytable.PrettyTable(['Producto', 'Mín', 'Máx'])
|
||||
table.align['Producto'] = 'l'
|
||||
table.align['Mín'] = 'r'
|
||||
table.align['Máx'] = 'r'
|
||||
for product in products:
|
||||
table.add_row([helpers.telegram_escape_characters(product['product_name']), f"{helpers.telegram_escape_characters(product['min_price'])}€", f"{helpers.telegram_escape_characters(product['max_price'])}€"])
|
||||
await update.message.reply_markdown_v2(f'```{(table)}```')
|
||||
|
||||
async def admin_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
if helpers.is_user_admin(update.message.chat_id):
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters("¡Eres admin!"))
|
||||
|
||||
async def add_user_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
async def add_premium_user_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
if helpers.is_user_admin(update.message.chat_id):
|
||||
telegram_user_id = update.message.text.split('/add_user ')[1]
|
||||
if walladb.add_valid_user(telegram_user_id):
|
||||
telegram_user_id = update.message.text.split('/add_premium_user ')[1]
|
||||
days = update.message.text.split('/add_premium_user ')[2]
|
||||
until = helpers.get_date_ahead(days)
|
||||
if walladb.add_premium_user(telegram_user_id, until):
|
||||
products = walladb.get_products_from_user(telegram_user_id)
|
||||
|
||||
for product in products:
|
||||
logging.info(product)
|
||||
p = threading.Thread(target=Worker.run, args=(product, ))
|
||||
p.start()
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} re-activado. Re-activando productos."))
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} re-activado hasta {until}. Re-activando productos."))
|
||||
else:
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} añadido."))
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} añadido hasta {until}."))
|
||||
|
||||
async def remove_user_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
if helpers.is_user_admin(update.message.chat_id):
|
||||
@@ -137,6 +155,26 @@ async def remove_user_command(update: Update, context: ContextTypes.DEFAULT_TYPE
|
||||
walladb.remove_valid_user(telegram_user_id)
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} desactivado."))
|
||||
|
||||
async def status_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
# TODO: Obtener estado del usuario y fecha de caducidad de suscripción en el caso de haberla
|
||||
message = ''
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
|
||||
|
||||
async def test_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
# TODO: Activar usuario en modo testing y responder con fecha de caducidad
|
||||
telegram_user_id = helpers.get_telegram_user_id(update)
|
||||
if not walladb.is_user_valid(telegram_user_id):
|
||||
until = helpers.get_date_ahead(7)
|
||||
walladb.add_test_user(telegram_user_id, until)
|
||||
message = f"Periodo de prueba activado hasta el {until}."
|
||||
else:
|
||||
message = "Ya has utilizado el periodo de prueba."
|
||||
if walladb.is_user_testing(telegram_user_id):
|
||||
message = "Ya estás en el periodo de prueba."
|
||||
elif walladb.is_user_premium(telegram_user_id):
|
||||
message = "Ya eres premium. No puedes volver al periodo de prueba."
|
||||
await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
|
||||
|
||||
def main()->None:
|
||||
walladb.setup_db()
|
||||
products = walladb.get_all_products()
|
||||
@@ -156,8 +194,10 @@ def main()->None:
|
||||
application.add_handler(CommandHandler("remove", remove_command))
|
||||
application.add_handler(CommandHandler("list", list_command))
|
||||
application.add_handler(CommandHandler("admin", admin_command))
|
||||
application.add_handler(CommandHandler("add_user", add_user_command))
|
||||
application.add_handler(CommandHandler("add_premium_user", add_premium_user_command))
|
||||
application.add_handler(CommandHandler("remove_user", remove_user_command))
|
||||
application.add_handler(CommandHandler("status", status_command))
|
||||
application.add_handler(CommandHandler("test", test_command))
|
||||
|
||||
# on non command i.e message - echo the message on Telegram
|
||||
#application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))
|
||||
|
||||
Reference in New Issue
Block a user