def get_timetable(text): neis = neispy.SyncClient(force=True) AE = "B10" # 교육청 코드 SE = 7010536 # 학교 코드 YMD = get_ymd(text) try: grade_idx = text.index("학년") - 1 class_idx = text.index("반") - 1 grade_no = int(text[grade_idx]) class_no = int(text[class_idx]) except Exception as e: print(f'exception! {e}') return "학년 반 정보를 제대로 입력해주세요!" try: if YMD: timetable_info = neis.timeTable(schclass='his', ATPT_OFCDC_SC_CODE=AE, SD_SCHUL_CODE=SE, ALL_TI_YMD=YMD, GRADE=grade_no, CLRM_NM=class_no) else: timetable_info = neis.timeTable(schclass='his', ATPT_OFCDC_SC_CODE=AE, SD_SCHUL_CODE=SE, GRADE=grade_no, CLRM_NM=class_no) timetable = [i['ITRT_CNTNT'] for i in timetable_info.data] result = '' for index, item in enumerate(timetable): result += f'{index+1}교시 - {item}\n' return result except Exception as e: print(f'exception! {e}') return "해당 날짜의 시간표 정보가 없어요 ㅠㅠ"
def get_schedule(text): neis = neispy.SyncClient(force=True) AE = "B10" # 교육청 코드 SE = 7010536 # 학교 코드 YMD = get_ymd(text) try: schedule_info = neis.SchoolSchedule(ATPT_OFCDC_SC_CODE=AE, SD_SCHUL_CODE=SE, AA_YMD=YMD) schedule = schedule_info.EVENT_NM return schedule except Exception as e: print(f'exception! {e}') return "해당 날짜의 학사일정 정보가 없어요 ㅠㅠ"
def get_meal(text): neis = neispy.SyncClient(force=True) AE = "B10" # 교육청 코드 SE = 7010536 # 학교 코드 YMD = get_ymd(text) try: meal_info = neis.mealServiceDietInfo(ATPT_OFCDC_SC_CODE=AE, SD_SCHUL_CODE=SE, MLSV_YMD=YMD) meal = meal_info.DDISH_NM.replace('<br/>', '\n') return meal except Exception as e: print(f'exception! {e}') return "해당 날짜의 급식정보가 없어요 ㅠㅠ"
load_dotenv(verbose=True) parser = argparse.ArgumentParser(description='Sunrinbot based on KoGPT-2') parser.add_argument('--model_params', type=str, default='model_chp/model_last.ckpt') args = parser.parse_args() app = Flask(__name__) ACCESS_TOKEN = os.getenv('ACCESS_TOKEN') VERIFY_TOKEN = os.getenv('VERIFY_TOKEN') bot = Bot(ACCESS_TOKEN) SCHOOL_NAME = "선린인터넷고등학교" neis = neispy.SyncClient(force=True) AE = "B10" # 교육청 코드 SE = 7010536 # 학교 코드 model = KoGPT2Chat.load_from_checkpoint(args.model_params) @app.route('/', methods=['GET', 'POST']) def chat(): res = "" if request.method == 'GET': token_sent = request.args.get('hub.verify_token') return vertify_token(token_sent) else: output = request.get_json() for event in output['entry']: