/
cli.py
187 lines (153 loc) · 5.93 KB
/
cli.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
# -*- coding: utf-8 -*-
import itertools
import json
import citybikes
import click
import geocoder
import colorama
from iso3166 import countries
__version__ = '0.1.9'
client = citybikes.Client()
network_name_template = u'Found network: {0[name]}, {1[city]} ({1[country]})'
country_name_template = u'{} [{}]'
epilog = ''.join([
"""\b
Brought to you by the fine folks at Citybikes.
""", colorama.Fore.GREEN,
"""\b
cccccccccccccccccccccccccccccccccccccccccc
ccccccccccccccccccccc cccccccc cccccc
ccccccccccccccccccccc ccccccc ccccc
cccccccccc cccccc ccc ccccc
cccccccccc ccc cccc
cccccccccc cccc
cccccccccc cccc ccc
ccccc cc cc c
ccccc ccc c
ccccc cc ccc c
ccccc ccccccc cccccccc c
c cc cccc cccccc c
c ccccc cc ccc c
c cccccccc cc ccc ccccccc c
c ccc ccccc cccc ccccc cc c
c ccc cc ccc cc cc cc cc c
c ccc cc cccc c cc c
c ccc ccc ccccccccccc c
c cccccccc cccccc c
c c
ccccccccccccccccccccccc https://citybik.es
""", colorama.Style.RESET_ALL])
def get_color(number):
if number > 5:
return 'green'
elif number > 1:
return 'yellow'
else:
return 'red'
def format_address(loc):
keys = [
# Stupid ping pong to get readable address out of bike station osm
# nodes. Nominatim seems to ignore osm_type query, which would make
# things easier
lambda addr: addr.get('address29') or addr.get('road'),
'house_number',
'postcode',
'city'
]
address = loc.raw['address']
components = [k(address) if callable(k) else address.get(k) for k in keys]
return ", ".join(filter(None, components))
def display_station(station, geocode=False, use_colors=False):
bar_width = 30
padding = ' '
bikes = int(station.get('free_bikes', 0))
slots = int(station.get('empty_slots', 0))
if 'slots' in station.get('extra', {}):
total_slots = float(station['extra']['slots'])
else:
total_slots = float(bikes + slots)
bikes_s = '%s bikes' % bikes
slots_s = '%s slots' % slots
try:
perc = int((bikes/total_slots) * bar_width)
except ZeroDivisionError:
perc = 0
status_bar = u'█' * perc + u'░' * (bar_width - perc)
status_pad = (bar_width - len(bikes_s) - len(slots_s)) * ' '
status = u''.join([
click.style(bikes_s, fg=get_color(bikes)),
status_pad,
click.style(slots_s, fg=get_color(slots))
])
if geocode:
loc = geocoder.osm([station['latitude'], station['longitude']],
method='reverse')
address = format_address(loc)
else:
address = 'lat, lng: {:.6f}, {:.6f}'.format(station['latitude'],
station['longitude'])
click.echo(padding + station['name'].title(), color=use_colors)
click.echo(click.style(padding + status_bar, fg=get_color(bikes)),
color=use_colors)
click.echo(click.style(padding + status), color=use_colors)
click.echo(padding + address)
@click.group(epilog=epilog)
def cli():
"""A command line client for the Citybikes API."""
pass
@click.command()
@click.argument('address', default=None)
@click.option('--geocode / --no-geocode', is_flag=True, default=True,
help='Geocode station positions into a readable address')
@click.option('-n', default=5, help='Number of stations to show.', type=int)
@click.option('--color / --no-color', is_flag=True, default=True,
help='Use colors on output')
@click.option('--json', 'output_json', is_flag=True, default=False,
help='Return JSON representation')
def show(address, geocode, n, color, output_json):
"""Display status of station on a given address."""
lat, lng = geocoder.osm(address).latlng
network, distance = next(iter(client.networks.near(lat, lng)))
if n == 0 or n > 10:
click.echo('Disabling geocoder, too many stations to geocode',
err=True, color=color)
geocode = False
if n == 0:
n = len(network.stations)
stations = [s for s, d in network.stations.near(lat, lng)][:n]
net_name = network_name_template.format(network, network['location'])
click.echo(click.style(net_name, fg='green'), err=True, color=color)
if output_json:
click.echo(json.dumps(stations, cls=citybikes.resource.JSONEncoder,
indent=4))
return
for station in stations:
display_station(station, geocode=geocode, use_colors=color)
click.echo()
@click.command()
def ls():
"""List all bike sharing networks supported."""
def get_ccode(network):
return network['location']['country']
def get_country(code):
try:
return countries.get(code.lower()).name
except KeyError:
return code
# Create iter of (country, network)
networks = ((get_country(get_ccode(n)), n) for n in client.networks)
# Sort networks by country
s_networks = sorted(networks, key=lambda cn: cn[0])
# Group networks by country
g_networks = itertools.groupby(s_networks, key=lambda cn: cn[0])
for country, networks in g_networks:
networks = [n for _, n in networks]
click.echo(country_name_template.format(country, len(networks)))
for i, n in enumerate(networks):
name = u'{0[location][city]} ({0[name]})'.format(n)
click.echo((u'├' if i < len(networks)-1 else u'└') + ' ' + name)
click.echo()
cli.add_command(show)
cli.add_command(ls)
if __name__ == '__main__':
cli()