北京医保定点医院

北京医保定点医院.csv

北京医保定点医院.py

#!/usr/bin/env python2
# coding: utf-8

import csv
import lxml.html
import prettytable
import requests

type_A = u'''\
中国医学科学院北京协和医院
首都医科大学附属北京同仁医院
首都医科大学宣武医院
首都医科大学附属北京友谊医院
北京大学第一医院(北京大学北大医院）
北京大学人民医院（北京大学第二临床医学院）
北京大学第三医院（北京大学第三临床医学院）
北京积水潭医院(北京大学第四临床医学院）
中国中医科学院广安门医院（中国中医科学院第二临床医药研究所）
首都医科大学附属北京朝阳医院
中日友好医院
北京大学首钢医院
首都医科大学附属北京中医医院
首都医科大学附属北京天坛医院
首都医科大学附属北京世纪坛医院（北京铁路总医院）
北京市健宫医院
北京市房山区良乡医院
北京市大兴区人民医院
北京市石景山医院'''.split('\n')
table = prettytable.PrettyTable(( u'编码', u'医院名称', u'所属区县', u'医院类别', u'医院等级', u'医保分类' ))
result = []

for sno in xrange( 0, 2080, 20 ):
    print 'fetching', sno
    resp = requests.get(
        'http://www.bjrbj.gov.cn/LDJAPP/search/ddyy/ddyy_01_outline.jsp',
        params={ 'sno':sno },
        allow_redirects=False,
    )
    assert resp.status_code == 200
    for tr in list( lxml.html.fromstring( resp.text ).cssselect('table table')[-1] )[1:-2]:
        assert len( tr ) == 5
        row = tuple( td[0].text_content() for td in tr )
        if row[1] in type_A:
            type_A.remove( row[1] )
            row += (u'A类',)
        elif u'中医' in row[3]:
            row += (u'中医',)
        elif u'专科' in row[3]:
            row += (u'专科',)
        else:
            row += ('',)
        table.add_row( row )
        result.append( row )

print table
print type_A

writer = csv.writer( open( '定点医院.csv', 'w' ))
writer.writerow([ s.encode('utf-8') for s in table.field_names ])
writer.writerows([ [cell.encode('utf-8') for cell in row] for row in result ])