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…
	
	Add table
		Add a link
		
	
		Reference in a new issue