class Ticket(object):
def __init__(self, config_file):
## config parser setting
self.config_file = config_file
self.settings = configparser.ConfigParser()
self.settings._interpolation = configparser.ExtendedInterpolation()
self.settings.read(self.config_file,'utf-8')
## environment setting
self.brower='chrome'
self.b = Browser(driver_name=self.brower)
self.station={}
self.url = "https://kyfw.12306.cn/otn/leftTicket/init"
# 席别类型(对应列标号)
self.ticket_index = [
'',
u'商务座',
u'特等座',
u'一等座',
u'二等座',
u'高级软卧',
u'软卧',
u'硬卧',
u'软座',
u'硬座',
u'无座'
]
self.username = ''
self.password = ''
self.date_format='%Y-%m-%d'
self.tolerance = -1
self.blacklist = {}
self.date = []
self.isStudent = False
self.success = 0
self.find_ticket = 0
self.config_parser()
self.playmusic = False
self.count = 0
def ConfigSectionMap(self,section):
dict1 = {}
options = self.settings.options(section)
for option in options:
try:
dict1[option] = self.settings.get(section, option)
if dict1[option] == -1:
DebugPrint("skip: %s" % option)
except:
print("exception on %s!" % option)
dict1[option] = None
return dict1
def daterange(self, start_date, end_date):
for n in range(int ((end_date - start_date).days) + 1):
yield start_date + timedelta(n)
def config_parser(self):
if self.retrieve_station_dict() == -1:
sys.exit()
if self.retrieve_book_options() == -1:
sys.exit()
def retrieve_station_dict(self):
dict_helper=self.ConfigSectionMap('STATIONCOOKIE')
for name, value in dict_helper.iteritems():
self.station[name]=value
def retrieve_book_options(self):
login_info=self.ConfigSectionMap('GLOBAL')
self.username = login_info['username'].strip()
self.password = login_info['password'].strip()
self.brower = login_info['browser']
book_settings = self.ConfigSectionMap('TICKET')
self.fromStation = [ station.strip() for station in book_settings['from_station'].split(',')]
self.toStation = [ station.strip() for station in book_settings['to_station'].split(',')]
trains = [ train.strip() for train in book_settings['trains'].split(',')]
if len(trains) == 1 and trains[0] == '':
self.trains = []
else:
self.trains = trains
self.ticket_type =[ _type.strip() for _type in book_settings['ticket_type'].split(',')]
rangeQuery = book_settings['range_query'].strip()
if rangeQuery == 'Y':
date = [ d.strip() for d in book_settings['date'].split(',')]
if len(date) < 2:
print "未设置正确的起至时间"
return -1
else:
start_date = datetime.strptime(date[0],self.date_format)
end_date = datetime.strptime(date[1],self.date_format)
if end_date < start_date:
print "查询截止日期不可大于开始日期!"
return -1
for single_date in self.daterange(start_date, end_date):
self.date.append(single_date.strftime(self.date_format))
else:
self.date = [ d.strip() for d in book_settings['date'].split(',')]
if book_settings['student'].strip() == 'Y':
self.isStudent = True
self.tolerance = int(book_settings['tolerance'])
self.people = [ people.strip() for people in book_settings['people'].split(',') ]
if book_settings['alarm'].strip() == 'Y':
self.playmusic = True
def login(self):
self.b.visit(self.url)
self.b.find_by_text(u"登录").click()
self.b.fill("loginUserDTO.user_name",self.username)
self.b.fill("userDTO.password",self.password)
pdb.set_trace()
def page_has_loaded(self):
page_state = self.b.evaluate_script("document.readyState")
return page_state == 'complete'
def switch_to_order_page(self):
order = []
while len(order) == 0:
order = self.b.find_by_text(u"车票预订")
order[0].click()
def checkTicket(self, date, fromStation, toStation):
print 'date: %s, from %s, to %s'%(date, fromStation, toStation)
self.b.cookies.add({"_jc_save_fromDate":date})
self.b.cookies.add({"_jc_save_fromStation":self.station[fromStation]})
self.b.cookies.add({"_jc_save_toStation":self.station[toStation]})
self.b.cookies.all()
self.b.reload()
if self.isStudent:
self.b.find_by_text(u'学生').click()
self.b.find_by_text(u"查询").click()
all_trains = []
while self.page_has_loaded() == False:
continue
while len(all_trains) == 0:
all_trains = self.b.find_by_id('queryLeftTable').find_by_tag('tr')
for k, train in enumerate(all_trains):
tds = train.find_by_tag('td')
if tds and len(tds) >= 10:
has_ticket= tds.last.find_by_tag('a')
if len(has_ticket) != 0:
if k + 1 < len(all_trains):
this_train = all_trains[k+1]['datatran']
if len(self.trains) != 0 and this_train not in self.trains:
continue
if self.tolerance != -1 and this_train in self.blacklist and self.blacklist[this_train] >= self.tolerance:
print u"%s 失败 %d 次, 跳过"%(this_train, self.blacklist[this_train])
continue
for cat in self.ticket_type:
if cat in self.ticket_index:
i = self.ticket_index.index(cat)
else:
print '无效的席别信息'
return 0, ''
if tds[i].text != u'无' and tds[i].text != '--':
if tds[i].text != u'有':
print u'%s 的 %s 有余票 %s 张!'%(this_train, cat ,tds[i].text)
else:
print u'%s 的 %s 有余票若干张!'%(this_train, cat)
self.find_ticket = 1
tds.last.click()
break
if self.find_ticket:
break
return this_train
def book(self):
while self.page_has_loaded() == False:
continue
if len(self.people) == 0:
print '没有选择乘车人!'
return 1
more = []
while len(more) == 0:
more = self.b.find_by_text(u"更多")
more[0].click()
person = []
people = self.people
while len(person) == 0:
person=self.b.find_by_xpath('//ul[@id="normal_passenger_id"]/li/label[contains(text(),"%s")]'%people[0])
for p in people:
self.b.find_by_xpath('//ul[@id="normal_passenger_id"]/li/label[contains(text(),"%s")]'%p).click()
if self.b.find_by_xpath('//div[@id="dialog_xsertcj"]').visible:
self.b.find_by_xpath('//div[@id="dialog_xsertcj"]/div/div/div/a[text()="确认"]').click()
return 1
def ring(self):
pygame.mixer.pre_init(64000, -16, 2, 4096)
pygame.init()
pygame.display.init()
screen=pygame.display.set_mode([300,300])
#pygame.display.flip()
pygame.time.delay(1000)#等待1秒让mixer完成初始化
tracker=pygame.mixer.music.load("media/sound.ogg")
#track = pygame.mixer.music.load("sound.ogg")
pygame.mixer.music.play()
# while pygame.mixer.music.get_busy():
#pygame.time.Clock().tick(10)
running = True
img=pygame.image.load("media/img.jpg")
while running:
screen.blit(img,(0,0))
pygame.display.flip()
for event in pygame.event.get():
if event.type==pygame.QUIT:
running = False
pygame.quit ()
return 1
def executor(self):
self.login()
self.switch_to_order_page()
while self.success == 0:
self.find_ticket = 0
while self.find_ticket == 0:
for date in self.date:
try:
self.count += 1
print "Try %d times" % self.count
for fromStation in self.fromStation:
for toStation in self.toStation:
this_train = self.checkTicket(date, fromStation, toStation)
if self.find_ticket:
break
if self.find_ticket:
break
if self.find_ticket:
break
except KeyboardInterrupt:
self.b.find_by_text(u'退出').click()
sys.exit()
except:
continue
# book ticket for target people
self.find_ticket = 0
while self.find_ticket == 0:
try:
self.find_ticket = self.book()
except KeyboardInterrupt:
self.b.find_by_text(u'退出').click()
sys.exit()
except:
continue
if self.playmusic:
self.ring()
print "订票成功了吗?(Y/N)"
input_var = ''
while input_var == '':
input_var= sys.stdin.read(1)
if input_var == 'Y' or input_var == 'y':
self.success = 1
elif input_var == 'N' or input_var == 'n':
if this_train in self.blacklist:
self.blacklist[this_train] += 1
else:
self.blacklist[this_train] = 1
print u"%s 失败 %d 次"%(this_train, self.blacklist[this_train])
self.b.back()
self.b.reload()
else:
input_var = ''
continue
self.b.find_by_text(u'退出').click()
def handle(self, driver, row, page, *args, **kwargs):
browser = Browser(driver)
browser.visit('http://www.dataescolabrasil.inep.gov.br/dataEscolaBrasil/')
with browser.get_iframe(0) as iframe:
iframe.check('situacaoEmAtividade')
iframe.select('estadoDecorate:estadoSelect', '41')
assert iframe.is_text_present('Abatia', wait_time=5) # wait a known city
iframe.find_by_id('pesquisar').first.click()
assert iframe.is_text_present('Foram encontradas', wait_time=5)
if page > 1:
next_page = 1
def click_pager():
scroller = iframe.find_by_id('resultadoDataScroller').first
pager = scroller.find_by_xpath('//td[text()="{0}"]'.format(next_page))
pager.first.click()
while 1:
try:
click_pager()
except ElementDoesNotExist:
try:
print 'retry', next_page
os.system('sleep 1s')
click_pager()
except ElementDoesNotExist:
print 're-retry', next_page
os.system('sleep 2s')
click_pager()
os.system('sleep 2s')
if next_page == page:
break
next_page += 1
while 1:
# Reload the trs to avoid outdated references
trs = iframe.find_by_css('#resultado tbody tr')
# Go to the next page when done
if (row + 1) > len(trs):
page += 1
row = 0
scroller = iframe.find_by_id('resultadoDataScroller').first
pager = scroller.find_by_xpath('//td[text()="{0}"]'.format(page))
pager.first.click()
continue
# Click first td anchor and assert it is the detail tab
tr = trs[row]
tds = tr.find_by_css('td')
tds[0].find_by_css('a').click()
if not iframe.is_text_present('Dados gerais'):
browser.back()
row += 1
iframe.find_by_xpath('//td[text()="Resultado de busca"]').first.click()
continue
# Get tab content
content = iframe.find_by_css('.rich-tabpanel-content')
data = self.data_from_ids(content, {
'external_code': 'dad_codEntidadeDecorate:dad_codEntidade',
'name': 'dad_noEntidadeDecorate:dad_noEntidade',
'city': 'dad_noMunicipioDecorate:dad_noMunicipio',
'address': 'dad_enderecoDecorate:dad_endereco',
'number': 'dad_numeroDecorate:dad_numero',
'complement': 'dad_complementoDecorate:dad_complemento',
'quarter': 'dad_bairroDecorate:dad_bairro',
'postal_code': 'dad_cepDecorate:dad_cep',
'phones': 'dad_numTelefoneDecorate:dad_numTelefone',
'email': 'dad_emailDecorate:dad_email',
'cnpj': 'cnpjEpDecorate:cnpjEp',
'adm_structure': 'dad_dependenciaAdmDecorate:dad_dependenciaAdm',
'private_school_category': 'descCategoriaEscPrivadaDecorate:descCategoriaEscPrivada',
})
try:
self.save_data(data, row=row, page=page)
except IntegrityError:
pass # could not save because cnpj already exists
# Go back and move to next row
iframe.find_by_xpath('//td[text()="Resultado de busca"]').first.click()
row += 1
class Ticket(object):
def __init__(self, config_file):
## config parser setting
self.config_file = config_file
self.settings = configparser.ConfigParser()
self.settings._interpolation = configparser.ExtendedInterpolation()
self.settings.read(self.config_file)
## environment setting
self.brower='chrome'
self.b = Browser(driver_name=self.brower)
self.station={}
self.url = "https://kyfw.12306.cn/otn/leftTicket/init"
# 席别类型(对应列标号)
self.ticket_index = [
'',
u'商务座',
u'特等座',
u'一等座',
u'二等座',
u'高级软卧',
u'软卧',
u'硬卧',
u'软座',
u'硬座',
u'无座'
]
self.username = ''
self.password = ''
self.date_format='%Y-%m-%d'
self.tolerance = -1
self.blacklist = {}
self.date = []
self.isStudent = False
self.success = 0
self.find_ticket = 0
self.config_parser()
self.playmusic = False
self.count = 0
def ConfigSectionMap(self,section):
dict1 = {}
options = self.settings.options(section)
for option in options:
try:
dict1[option] = self.settings.get(section, option)
if dict1[option] == -1:
DebugPrint("skip: %s" % option)
except:
print("exception on %s!" % option)
dict1[option] = None
return dict1
def daterange(self, start_date, end_date):
for n in range(int ((end_date - start_date).days) + 1):
yield start_date + timedelta(n)
def config_parser(self):
if self.retrieve_station_dict() == -1:
sys.exit()
if self.retrieve_book_options() == -1:
sys.exit()
def retrieve_station_dict(self):
dict_helper=self.ConfigSectionMap('STATIONCOOKIE')
for name, value in dict_helper.iteritems():
self.station[name]=value
def retrieve_book_options(self):
login_info=self.ConfigSectionMap('GLOBAL')
self.username = login_info['username'].strip()
self.password = login_info['password'].strip()
self.brower = login_info['browser']
book_settings = self.ConfigSectionMap('TICKET')
self.fromStation = [ station.strip() for station in book_settings['from_station'].split(',')]
self.toStation = [ station.strip() for station in book_settings['to_station'].split(',')]
trains = [ train.strip() for train in book_settings['trains'].split(',')]
if len(trains) == 1 and trains[0] == '':
self.trains = []
else:
self.trains = trains
self.ticket_type =[ _type.strip() for _type in book_settings['ticket_type'].split(',')]
rangeQuery = book_settings['range_query'].strip()
if rangeQuery == 'Y':
date = [ d.strip() for d in book_settings['date'].split(',')]
if len(date) < 2:
print "未设置正确的起至时间"
return -1
else:
start_date = datetime.strptime(date[0],self.date_format)
end_date = datetime.strptime(date[1],self.date_format)
if end_date < start_date:
print "查询截止日期不可大于开始日期!"
return -1
for single_date in self.daterange(start_date, end_date):
self.date.append(single_date.strftime(self.date_format))
else:
self.date = [ d.strip() for d in book_settings['date'].split(',')]
if book_settings['student'].strip() == 'Y':
self.isStudent = True
self.tolerance = int(book_settings['tolerance'])
self.people = [ people.strip() for people in book_settings['people'].split(',') ]
if book_settings['alarm'].strip() == 'Y':
self.playmusic = True
def login(self):
self.b.visit(self.url)
self.b.find_by_text(u"登录").click()
self.b.fill("loginUserDTO.user_name",self.username)
self.b.fill("userDTO.password",self.password)
pdb.set_trace()
def page_has_loaded(self):
page_state = self.b.evaluate_script("document.readyState")
return page_state == 'complete'
def switch_to_order_page(self):
order = []
while len(order) == 0:
order = self.b.find_by_text(u"车票预订")
order[0].click()
def checkTicket(self, date, fromStation, toStation):
print 'date: %s, from %s, to %s'%(date, fromStation, toStation)
self.b.cookies.add({"_jc_save_fromDate":date})
self.b.cookies.add({"_jc_save_fromStation":self.station[fromStation]})
self.b.cookies.add({"_jc_save_toStation":self.station[toStation]})
self.b.cookies.all()
self.b.reload()
if self.isStudent:
self.b.find_by_text(u'学生').click()
self.b.find_by_text(u"查询").click()
all_trains = []
while self.page_has_loaded() == False:
continue
while len(all_trains) == 0:
all_trains = self.b.find_by_id('queryLeftTable').find_by_tag('tr')
for k, train in enumerate(all_trains):
tds = train.find_by_tag('td')
if tds and len(tds) >= 10:
has_ticket= tds.last.find_by_tag('a')
if len(has_ticket) != 0:
if k + 1 < len(all_trains):
this_train = all_trains[k+1]['datatran']
if len(self.trains) != 0 and this_train not in self.trains:
continue
if self.tolerance != -1 and this_train in self.blacklist and self.blacklist[this_train] >= self.tolerance:
print u"%s 失败 %d 次, 跳过"%(this_train, self.blacklist[this_train])
continue
for cat in self.ticket_type:
if cat in self.ticket_index:
i = self.ticket_index.index(cat)
else:
print '无效的席别信息'
return 0, ''
if tds[i].text != u'无' and tds[i].text != '--':
if tds[i].text != u'有':
print u'%s 的 %s 有余票 %s 张!'%(this_train, cat ,tds[i].text)
else:
print u'%s 的 %s 有余票若干张!'%(this_train, cat)
self.find_ticket = 1
tds.last.click()
break
if self.find_ticket:
break
return this_train
def book(self):
while self.page_has_loaded() == False:
continue
if len(self.people) == 0:
print '没有选择乘车人!'
return 1
more = []
while len(more) == 0:
more = self.b.find_by_text(u"更多")
more[0].click()
person = []
people = self.people
while len(person) == 0:
person=self.b.find_by_xpath('//ul[@id="normal_passenger_id"]/li/label[contains(text(),"%s")]'%people[0])
for p in people:
self.b.find_by_xpath('//ul[@id="normal_passenger_id"]/li/label[contains(text(),"%s")]'%p).click()
if self.b.find_by_xpath('//div[@id="dialog_xsertcj"]').visible:
self.b.find_by_xpath('//div[@id="dialog_xsertcj"]/div/div/div/a[text()="确认"]').click()
return 1
def ring(self):
pygame.mixer.pre_init(64000, -16, 2, 4096)
pygame.init()
pygame.display.init()
screen=pygame.display.set_mode([300,300])
#pygame.display.flip()
pygame.time.delay(1000)#等待1秒让mixer完成初始化
tracker=pygame.mixer.music.load("media/sound.ogg")
#track = pygame.mixer.music.load("sound.ogg")
pygame.mixer.music.play()
# while pygame.mixer.music.get_busy():
#pygame.time.Clock().tick(10)
running = True
img=pygame.image.load("media/img.jpg")
while running:
screen.blit(img,(0,0))
pygame.display.flip()
for event in pygame.event.get():
if event.type==pygame.QUIT:
running = False
pygame.quit ()
return 1
def executor(self):
self.login()
self.switch_to_order_page()
while self.success == 0:
self.find_ticket = 0
while self.find_ticket == 0:
for date in self.date:
try:
self.count += 1
print "Try %d times" % self.count
for fromStation in self.fromStation:
for toStation in self.toStation:
this_train = self.checkTicket(date, fromStation, toStation)
if self.find_ticket:
break
if self.find_ticket:
break
if self.find_ticket:
break
except KeyboardInterrupt:
self.b.find_by_text(u'退出').click()
sys.exit()
except:
continue
# book ticket for target people
self.find_ticket = 0
while self.find_ticket == 0:
try:
self.find_ticket = self.book()
except KeyboardInterrupt:
self.b.find_by_text(u'退出').click()
sys.exit()
except:
continue
if self.playmusic:
self.ring()
print "订票成功了吗?(Y/N)"
input_var = ''
while input_var == '':
input_var= sys.stdin.read(1)
if input_var == 'Y' or input_var == 'y':
self.success = 1
elif input_var == 'N' or input_var == 'n':
if this_train in self.blacklist:
self.blacklist[this_train] += 1
else:
self.blacklist[this_train] = 1
print u"%s 失败 %d 次"%(this_train, self.blacklist[this_train])
self.b.back()
self.b.reload()
else:
input_var = ''
continue
self.b.find_by_text(u'退出').click()
