134 lines
3.7 KiB
Python
134 lines
3.7 KiB
Python
import datetime
|
|
import dateutil
|
|
import logging
|
|
import json
|
|
import re
|
|
import requests
|
|
|
|
|
|
def init(CONFIG, logger):
|
|
logger.setLevel(logging.DEBUG)
|
|
return Mod(CONFIG, logger)
|
|
|
|
|
|
class Mod:
|
|
def __init__(self, CONFIG, logging):
|
|
self.URL = "https://www.e-license.jp/el25/pc/"
|
|
self.CONFIG = CONFIG
|
|
|
|
self.logging = logging
|
|
self.targets = []
|
|
self.calender = None
|
|
|
|
self.ss = self.authenticate()
|
|
|
|
def cycle(self):
|
|
if len(self.targets) == 0:
|
|
return
|
|
|
|
res = self.ss.post(
|
|
url = self.URL+"p04a.action",
|
|
data = {
|
|
"b.processCd": "A",
|
|
"b.kamokuCd" : 0,
|
|
"b.page" : 1,
|
|
"b.schoolCd" : self.CONFIG["school"],
|
|
})
|
|
res.encoding = "shift_jis"
|
|
self.calender = Calender(res.text)
|
|
|
|
now = datetime.datetime.utcnow() + datetime.timedelta(hours=9)
|
|
today = now.strftime("%Y%m%d")
|
|
|
|
for target in self.targets:
|
|
for slot in target[1]:
|
|
begins = now.combine(now.date(), TIMETABLE[slot])
|
|
border = begins - datetime.timedelta(hours=2)
|
|
|
|
if target[0] != today or now < border:
|
|
if self.calender.checkAvailable(target[0], slot):
|
|
self.make_reserve(target[0], slot)
|
|
target[2]()
|
|
break
|
|
self.targets = []
|
|
|
|
def watch(self, date, slots, cb):
|
|
self.logging.debug("watch request accepted, {} {}".format(date, slots))
|
|
self.targets.append((date, slots, cb))
|
|
|
|
def make_reserve(self, date, slot):
|
|
res = self.ss.post(
|
|
url = self.URL+"p03a.action",
|
|
data = {
|
|
"b.schoolCd" : self.CONFIG["school"],
|
|
"b.processCd" : "V",
|
|
"b.kamokuCd" : 0,
|
|
"b.instructorTypeCd" : 0,
|
|
"b.dateInformationType": date,
|
|
"b.infoPeriodNumber" : slot,
|
|
"b.carModelCd" : self.calender.carModel,
|
|
"b.instructorCd" : 0,
|
|
"b.page" : 1,
|
|
"b.groupCd" : self.calender.group,
|
|
})
|
|
res.encoding = "shift_jis"
|
|
self.calender = Calender(res.text)
|
|
|
|
self.logging.info("reservation made, {}:{}".format(date, slots))
|
|
|
|
def authenticate(self):
|
|
ss = requests.Session()
|
|
try:
|
|
res = ss.post(
|
|
url = self.URL+"p01a.action",
|
|
data = {
|
|
"b.studentId" : self.CONFIG["username"],
|
|
"b.password" : self.CONFIG["password"],
|
|
"b.schoolCd" : self.CONFIG["school"],
|
|
"b.processCd" : "",
|
|
"b.kamokuCd" : "",
|
|
"method:doLogin" : "ログイン".encode("shift_jis"),
|
|
})
|
|
res.encoding = "shift_jis"
|
|
self.calender = Calender(res.text)
|
|
self.logging.info("authenticated")
|
|
return ss
|
|
except Exception as e:
|
|
self.logging.error("authentication failure ({})".format(e))
|
|
return None
|
|
|
|
|
|
class Calender:
|
|
def __init__(self, text):
|
|
m = re.search(RE_ERROR, text)
|
|
if m is None:
|
|
m = re.search(RE_ALERT, text)
|
|
if m is not None:
|
|
raise Exception(m[1])
|
|
|
|
self.avails = re.findall(RE_AVAIL_SLOT, text)
|
|
|
|
def checkAvailable(self, date, slot):
|
|
for avail in self.avails:
|
|
if avail[0] == date and avail[1] == str(slot):
|
|
return True
|
|
return False
|
|
|
|
|
|
TIMETABLE = {
|
|
1: datetime.time( 9,20),
|
|
2: datetime.time(10,20),
|
|
3: datetime.time(11,20),
|
|
4: datetime.time(12,20),
|
|
5: datetime.time(14,10),
|
|
6: datetime.time(15,10),
|
|
7: datetime.time(16,10),
|
|
8: datetime.time(17,10),
|
|
9: datetime.time(18,10),
|
|
10: datetime.time(19,10),
|
|
}
|
|
|
|
RE_ALERT = re.compile(r"window\.alert\('(.*)'\)")
|
|
RE_ERROR = re.compile(r"<.*? class=\"errorTitle\">メッセージ<\/.*?>\s*<.*? class=\"errorDisp\">(.*?)<\/.*?>", re.MULTILINE)
|
|
RE_AVAIL_SLOT = re.compile(r"^\s*<a href=\"#\" onclick=\"sendContent\('(\d+)','(\d+)','V',document\.getElementById\('formId'\)\)\">$")
|