Added first functional code
ability to insert a station_id via inline message; ability to display departures for specified station_name or station_id in direct chat
This commit is contained in:
parent
87b2b5c875
commit
ccc9577716
1 changed files with 74 additions and 0 deletions
74
bot.py
Normal file
74
bot.py
Normal file
|
@ -0,0 +1,74 @@
|
|||
import logging
|
||||
import requests
|
||||
from telegram import InlineQueryResultArticle, InputTextMessageContent
|
||||
from telegram.ext import Updater, CommandHandler, InlineQueryHandler
|
||||
|
||||
token_file = open("token", "r")
|
||||
updater = Updater(token=token_file.read(), use_context=True)
|
||||
token_file.close()
|
||||
|
||||
dispatcher = updater.dispatcher
|
||||
|
||||
logging.basicConfig(format='%(acstime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
|
||||
|
||||
|
||||
def start(update):
|
||||
update.message.reply_text("Hello")
|
||||
|
||||
|
||||
def search_station(query):
|
||||
request = requests.get("https://efa-api.asw.io/api/v1/station/?search=" + query)
|
||||
if request.status_code != 200:
|
||||
return -1
|
||||
else:
|
||||
return request.json()[0]['stationId']
|
||||
|
||||
|
||||
def get_vvs_departures(update, context):
|
||||
if len(context.args) == 0:
|
||||
update.message.reply_text("No argument specified")
|
||||
return
|
||||
if context.args[0].isdigit():
|
||||
station_id = context.args[0]
|
||||
else:
|
||||
station_id = search_station(' '.join(context.args))
|
||||
request = requests.get("https://efa-api.asw.io/api/v1/station/" + station_id + "/departures")
|
||||
if request.status_code != 200:
|
||||
update.message.reply_text("Error with server communication")
|
||||
update.message.reply_text("Next departures:")
|
||||
for station in request.json()[:4]:
|
||||
update.message.reply_text(
|
||||
"Line " + station['number'] + " to \"" + station['direction'] + "\" at "
|
||||
+ station['departureTime']['hour'].zfill(2) + ':' + station['departureTime']['minute'].zfill(2)
|
||||
+ " +" + str(station['delay']))
|
||||
|
||||
|
||||
def inline_station_search(update, context):
|
||||
query = update.inline_query.query
|
||||
if len(query) < 5 or not query:
|
||||
return
|
||||
results = list()
|
||||
for station in get_stations_by_search_query(query):
|
||||
results.append(
|
||||
InlineQueryResultArticle(
|
||||
id=station['stationId'],
|
||||
title=station['name'],
|
||||
input_message_content=InputTextMessageContent("/vvs " + station['stationId'])
|
||||
)
|
||||
)
|
||||
context.bot.answer_inline_query(update.inline_query.id, results)
|
||||
|
||||
|
||||
def get_stations_by_search_query(name):
|
||||
request = requests.get("https://efa-api.asw.io/api/v1/station/?search=" + name)
|
||||
return request.json()
|
||||
|
||||
|
||||
inline_station_search_handler = InlineQueryHandler(inline_station_search)
|
||||
|
||||
dispatcher.add_handler(inline_station_search_handler)
|
||||
|
||||
updater.dispatcher.add_handler(CommandHandler('vvs', get_vvs_departures))
|
||||
|
||||
updater.start_polling()
|
||||
updater.idle()
|
Loading…
Reference in a new issue