Beispiel #1
0
def build_cache():
    import logging
    logging.basicConfig(level=logging.INFO)

    cache.invalidate(hard=True)
    BeijingBus.get_all_stations()

    click.secho('Done!', fg='green')
Beispiel #2
0
def build_cache():
    import logging
    logging.basicConfig(level=logging.INFO)

    cache.invalidate(hard=True)
    BeijingBus.get_all_stations()

    click.secho('Done!', fg='green')
def get_s2s_data(msg):
    stations = BeijingBus.extract_stations(msg)
    if len(stations) < 2:
        output = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return output

    from_station, to_station = stations[:2]
    lines = BeijingBus.extract_lines(msg)
    lines = match_stations_with_lines(from_station, to_station, lines)
    if not lines:
        output = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return output

    output = get_realtime_message(lines, from_station)
    return output
Beispiel #4
0
def query():
    q = click.prompt('请输入线路名', value_proc=str)
    lines = BeijingBus.search_lines(q)
    for index, line in enumerate(lines):
        click.echo()
        click.secho('[%s] %s' % (index + 1, line.name),
                    bold=True,
                    underline=True)
        station_names = [s.name for s in line.stations]
        click.echo()
        click.echo('站点列表:%s' % ','.join(station_names))

    click.echo()
    q = click.prompt('请从结果中选择线路编号', type=int)

    line = lines[q - 1]
    click.clear()
    click.echo('你选择了 %s,下面请选择站点' % line.name)
    click.echo()
    for index, station in enumerate(line.stations):
        click.echo('[%s] %s' % (index + 1, station.name))

    click.echo()
    q = click.prompt('请从结果中选择线路编号', type=int)

    while True:
        echo_realtime_data(line, q)
        time.sleep(5)
Beispiel #5
0
def cli():
    q = click.prompt('请输入线路名', value_proc=str)
    lines = BeijingBus.search_lines(q)
    for index, line in enumerate(lines):
        click.echo()
        click.secho('[%s] %s' % (index+1, line.name), bold=True, underline=True)
        station_names = [s.name for s in line.stations]
        click.echo()
        click.echo('站点列表:%s' % ','.join(station_names))

    click.echo()
    q = click.prompt('请从结果中选择线路编号', type=int)

    line = lines[q-1]
    click.clear()
    click.echo('你选择了 %s,下面请选择站点' % line.name)
    click.echo()
    for index, station in enumerate(line.stations):
        click.echo('[%s] %s' % (index+1, station.name))

    click.echo()
    q = click.prompt('请从结果中选择线路编号', type=int)

    while True:
        echo_realtime_data(line, q)
        time.sleep(5)
def get_lines(line_num):
    lines = BeijingBus.search_lines(line_num)
    result = ''
    for line_index, line in enumerate(lines):
        result += '[{}] {}\n'.format(line_index+1, line.name)
        station_names = [s.name for s in line.stations]
        '站点列表:{}\n\n'.format(','.join(station_names))
    return result + '请从结果中选择线路编号'
def get_line_info(line_query):
    result = ''
    lines = BeijingBus.search_lines(line_query[0])
    if line_query[1].isdigit() and string.atoi(line_query[1]) <= len(lines):
        line = lines[string.atoi(line_query[1])-1]
        for s_index, station in enumerate(line.stations):
            result += '[{}] {}\n'.format(s_index+1, station.name)
        result += '你选择了 {},下面请选择站点'.format(line.name)
    return result
Beispiel #8
0
def match_stations_with_lines(from_station, to_station, lines=None):
    def match(a, b, L):
        """检查L中包含a和b且a比b靠前"""
        try:
            return L.index(a) < L.index(b)
        except ValueError:
            return False

    if not lines:
        lines = BeijingBus.get_all_lines()

    return [line for line in lines if match(from_station, to_station, line.stations)]
Beispiel #9
0
def query(**kwargs):
    username = kwargs.get('sender')
    sender = kwargs.get('receiver')
    message_type = kwargs.get('type')

    def r(content):
        return weixin.reply(username, sender=sender, content=content)

    if message_type == 'event' and kwargs.get('event') == 'subscribe':
        msg = app.config.get('ON_FOLLOW_MESSAGE')
        if not msg:
            return ''

        return weixin.reply(username,
                            type='news',
                            sender=sender,
                            articles=[msg])

    content = kwargs.get('content')
    if not content:
        reply = '我好笨笨哦,还不懂你在说什么。\n%s' % QUERY_EXAMPLE
        return r(reply)

    if isinstance(content, unicode):
        content = content.encode('utf-8')

    stations = BeijingBus.extract_stations(content)
    lines = BeijingBus.extract_lines(content)
    if len(stations) < 2:
        reply = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return r(reply)

    from_station, to_station = stations[:2]
    lines = match_stations_with_lines(from_station, to_station, lines)
    if not lines:
        reply = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return r(reply)

    reply = get_realtime_message(lines, from_station)
    return r(reply)
Beispiel #10
0
def list_supported_lines():
    names = set([
        line.short_name for line in BeijingBus.get_all_lines()
    ])
    return '''<!DOCTYPE html>
              <html lang="zh-CN">
                <head>
                  <meta name="viewport" content="width=device-width, initial-scale=1.0">  
                  <title>支持的线路的列表</title>
                </head>
                <body>%s</body>
              </html>
           ''' % ''.join(['<p>%s</p>' % name for name in sorted(names)])
Beispiel #11
0
def query(**kwargs):
    username = kwargs.get('sender')
    sender = kwargs.get('receiver')
    message_type = kwargs.get('type')

    def r(content):
        return weixin.reply(
            username, sender=sender, content=content
        )

    if message_type == 'event' and kwargs.get('event') == 'subscribe':
        reply = ('欢迎关注北京实时公交!\n'
                 '你可以通过向我发送消息查询公交实时到站时间。\n\n%s'
                ) % QUERY_EXAMPLE
        return r(reply)

    content = kwargs.get('content')
    if not content:
        reply = '我好笨笨哦,还不懂你在说什么。\n%s' % QUERY_EXAMPLE
        return r(reply)

    if isinstance(content, unicode):
        content = content.encode('utf-8')

    stations = BeijingBus.extract_stations(content)
    lines = BeijingBus.extract_lines(content)
    if len(stations) < 2:
        reply = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return r(reply)

    from_station, to_station = stations[:2]
    lines = match_stations_with_lines(from_station, to_station, lines)
    if not lines:
        reply = '没有结果,可能还不支持这条线路呢~ \n%s' % QUERY_EXAMPLE
        return r(reply)

    reply = get_realtime_message(lines, from_station)
    return r(reply)
Beispiel #12
0
def query(**kwargs):
    username = kwargs.get("sender")
    sender = kwargs.get("receiver")
    message_type = kwargs.get("type")

    def r(content):
        return weixin.reply(username, sender=sender, content=content)

    if message_type == "event" and kwargs.get("event") == "subscribe":
        msg = app.config.get("ON_FOLLOW_MESSAGE")
        if not msg:
            return ""

        return weixin.reply(username, type="news", sender=sender, articles=[msg])

    content = kwargs.get("content")
    if not content:
        reply = "我好笨笨哦,还不懂你在说什么。\n%s" % QUERY_EXAMPLE
        return r(reply)

    if isinstance(content, unicode):
        content = content.encode("utf-8")

    stations = BeijingBus.extract_stations(content)
    lines = BeijingBus.extract_lines(content)
    if len(stations) < 2:
        reply = "没有结果,可能还不支持这条线路呢~ \n%s" % QUERY_EXAMPLE
        return r(reply)

    from_station, to_station = stations[:2]
    lines = match_stations_with_lines(from_station, to_station, lines)
    if not lines:
        reply = "没有结果,可能还不支持这条线路呢~ \n%s" % QUERY_EXAMPLE
        return r(reply)

    reply = get_realtime_message(lines, from_station)
    return r(reply)
Beispiel #13
0
def match_stations_with_lines(from_station, to_station, lines=None):

    def match(a, b, l):
        '''检查l中包含a和b且a比b靠前'''
        try:
            return l.index(a) < l.index(b)
        except ValueError:
            return False

    if not lines:
        lines = BeijingBus.get_all_lines()

    return [
        line for line in lines if match(from_station, to_station, line.stations)
    ]
def get_station_info(line_query):
    result = ''
    lines = BeijingBus.search_lines(line_query[0])
    if line_query[1].isdigit() and string.atoi(line_query[1]) <= len(lines):
        line = lines[string.atoi(line_query[1]) - 1]
        if line_query[2].isdigit() and string.atoi(line_query[2]) <= len(lines):
            station = line.stations[string.atoi(line_query[2]) - 1]
            realtime_data = line.get_realtime_data(string.atoi(line_query[2]))
            now = datetime.now().strftime('%H:%M:%S')
            result += '实时数据: {} 线路: {} 更新于: {}\n'.format(station.name, line.name, now)
            realtime_data = filter(lambda d: d['station_arriving_time'], realtime_data)
            realtime_data = sorted(realtime_data, key=lambda d: d['station_arriving_time'], reverse=False)
            for i, data in enumerate(realtime_data):
                result += '公交车辆 {}:\n'.format(i + 1)
                result += '距离 {} 还有 {} 米\n'.format(station.name, data['station_distance'])
                result += '预计 {} 到达\n'.format(data['station_arriving_time'].strftime('%H:%M'))

    result += '感谢使用!重新查询请重新输入线路。'
    return result
Beispiel #15
0
def build_cache():
    import logging
    logging.basicConfig(level=logging.DEBUG)
    BeijingBus.build_cache()
    click.secho('Done!', fg='green')
Beispiel #16
0
def build_cache():
    import logging
    logging.basicConfig(level=logging.DEBUG)
    BeijingBus.build_cache()
    click.secho('Done!', fg='green')
Beispiel #17
0
def list_supported_lines():
    names = set([line.short_name for line in BeijingBus.get_all_lines()])
    names = sorted([n.decode("utf-8") for n in names])
    return render_template("list.html", line_names=names)
Beispiel #18
0
def list_supported_lines():
    names = set([
        line.short_name for line in BeijingBus.get_all_lines()
    ])
    names = sorted([n.decode('utf-8') for n in names])
    return render_template('list.html', line_names=names)