버스 정류장 시간대별 승하차 인원 분석

get_buses.py

from datetime import datetime
from requests import post
import pymongo
import sys

# 8100: 234000878
# 8110: 204000082
# m4102: 234001159

def get_buses(bus_no):
    bus_dic = {"b8100": 234000878, "b8110": 204000082, "m4102": 234001159}
    user = ""
    passwd = ""
    db = ""
    client = pymongo.MongoClient('mongodb://{}:{}@127.0.0.1:27017/{}'.format(user, passwd, db))

    db = client.busDB

    url = "http://www.gbis.go.kr/gbis2014/schBusAPI.action"
    route_id = bus_dic[bus_no]
    data = post(url, data={"cmd": "searchRouteJson", "routeId": route_id}).json()
    buses = []
    for bus in data["result"]["realTime"]["list"]:
        bus.update(crawled_at=datetime.now())
        buses.append(bus)
    return db, buses