Allow custom default locations

This commit is contained in:
Davte 2019-08-15 20:08:19 +02:00
parent 42cf2db0af
commit dfa1dee5c3

View File

@ -306,11 +306,6 @@ class Location():
) )
default_location = Location(
(43.718518, 10.402165) # Borgo Stretto Station
)
class Station(Location): class Station(Location):
"""CicloPi bike sharing station.""" """CicloPi bike sharing station."""
@ -1631,8 +1626,21 @@ async def _ciclopi_button(bot, update, user_record):
return result return result
def init(bot, ciclopi_messages=None): def init(bot, ciclopi_messages=None, _default_location=(43.718518, 10.402165)):
"""Take a bot and assign commands to it.""" """Take a bot and assign CicloPi-related commands to it.
`ciclopi_messages` : dict
Multilanguage dictionary with all CicloPi-related messages.
`default_location` : tuple (float, float)
Tuple of coordinates (latitude, longitude) of default location.
Defaults to Borgo Stretto CicloPi station.
"""
# Define a global `default_location` variable holding default location
global default_location
default_location = Location(_default_location)
bot.ciclopi_default_location = default_location
with bot.db as db: with bot.db as db:
if 'ciclopi_stations' not in db.tables: if 'ciclopi_stations' not in db.tables:
db['ciclopi_stations'].insert_many( db['ciclopi_stations'].insert_many(