diff --git a/access.py b/access.py
new file mode 100644
index 0000000..050f3c5
--- /dev/null
+++ b/access.py
@@ -0,0 +1,88 @@
+#! /usr/bin/env python3
+
+# necessary modules
+import requests
+from bs4 import BeautifulSoup
+# configurations
+import config
+
+class kuser:
+ def __init__(self, un='', pw=''):
+ self.username = un
+ self.password = pw
+ self.session = requests.Session()
+ # login kCOJ
+ def login(self):
+ payload = {'name': self.username,
+ 'passwd': self.password,
+ 'rdoCourse': 1}
+ return self.session.post(config.URL + '/Login', data=payload)
+ # check online status
+ def check_online(self):
+ response = self.session.get(config.URL + '/TopMenu')
+ soup = BeautifulSoup(response.text, 'html.parser')
+ return soup.find('a').get_text() == '線上考試'
+ # list all questions, deadline and hand-in status
+ def list_questions(self):
+ questions = {}
+ response = self.session.get(config.URL + '/HomeworkBoard')
+ soup = BeautifulSoup(response.text, 'html.parser')
+ for tag in soup.find_all('tr'):
+ if tag.find('a') == None:
+ continue
+ else:
+ number = tag.find('a').get_text()
+ deadline = tag.find_all('td')[3].get_text()
+ status = tag.find_all('td')[5].get_text()
+ questions[number] = (deadline, status)
+ return questions
+ # show the content of the question
+ def show_question(self, number):
+ response = self.session.get(config.URL + '/showHomework', params={'hwId': number})
+ soup = BeautifulSoup(response.text, 'html.parser')
+ content = str(soup.find('body'))
+ content = content.replace('\n', '')
+ content = content.replace('\n\n', '').replace('\n', '')
+ content = content.replace('', '')
+ content = content.replace('', '').replace('
', '\n').replace('
', '\n').replace(' ', '')
+ return content
+ # list passers of the question
+ def list_passers(self, number):
+ passers = []
+ response = self.session.get(config.URL + '/success.jsp', params={'HW_ID': number})
+ soup = BeautifulSoup(response.text, 'html.parser')
+ for tr in soup.find_all('tr'):
+ passer = tr.get_text().replace('\n', '')
+ if passer != '學號':
+ passers += [passer]
+ return passers
+ def list_results(self, number):
+ results = []
+ response = self.session.get(config.URL + '/CheckResult.jsp', params={'questionID': number, 'studentID': self.username})
+ soup = BeautifulSoup(response.text, 'html.parser')
+ for tr in soup.find_all('tr'):
+ td = tr.find('td')
+ if td.get_text() != '測試編號':
+ results += [(td.get_text(), tr.find_all('td')[1].get_text())]
+ return results
+ # delete the answer of the question
+ def delete_answer(self, number):
+ response = self.session.get(config.URL + '/delHw', params={'title': number})
+ soup = BeautifulSoup(response.text, 'html.parser')
+ return soup.find('body').get_text().replace('\n', '') == 'delete success'
+ # hand in a answer
+ def upload_answer(self, number):
+ response = self.session.get(config.URL + '/upLoadHw', params={'hwId': number})
+ response = self.session.post(config.URL + '/upLoadFile')
+ return response.text # unavailable
+
+# for debug
+def main():
+ users = {}
+ users[config.DEBUG_ID] = kuser(config.DEBUG_USER, config.DEBUG_PSWD)
+ me = users[config.DEBUG_ID]
+ me.login()
+ print(me.upload_answer('012'))
+
+if __name__ == '__main__':
+ main()
\ No newline at end of file
diff --git a/bot.py b/bot.py
new file mode 100644
index 0000000..5136676
--- /dev/null
+++ b/bot.py
@@ -0,0 +1,36 @@
+#! /usr/bin/env python3
+
+# necessary modules
+import telepot, time, requests
+from telepot.loop import MessageLoop
+from pprint import pprint
+# kCOJ API
+import access
+# configurations
+import config
+
+def on_chat(msg):
+ pprint(msg)
+ content_type, chat_type, chat_id = telepot.glance(msg)
+
+ # me = users[str(chat_id)]
+ me = users[config.DEBUG_ID] # for debug
+ me.login()
+
+ if (msg['text'] in me.list_questions()):
+ content = me.show_question(msg['text'])
+ bot.sendMessage(chat_id, '' + content + '', parse_mode='HTML')
+ else:
+ bot.sendMessage(chat_id, '```\n' + str(msg) + '\n```', parse_mode='Markdown')
+
+users = {}
+bot = telepot.Bot(config.TOKEN)
+MessageLoop(bot, {'chat': on_chat}).run_as_thread()
+
+# for debug
+def main():
+ users[config.DEBUG_ID] = access.kuser(config.DEBUG_USER, config.DEBUG_PSWD)
+
+main()
+while True:
+ time.sleep(10)
\ No newline at end of file
diff --git a/config.py b/config.py
new file mode 100644
index 0000000..865b373
--- /dev/null
+++ b/config.py
@@ -0,0 +1,8 @@
+# kCOJ_bot
+TOKEN = ""
+# kCOJ
+URL = ""
+# debug config
+DEBUG_ID = ""
+DEBUG_USER = ""
+DEBUG_PSWD = ""
\ No newline at end of file