Merge pull request #17 from khw2ok/master

Update url address
Coder-Iro · Aug 7, 2024 · 306d677 · 306d677
2 parents b160eba + 3a31560
commit 306d677
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/comcigan/async_school.py b/comcigan/async_school.py
@@ -28,15 +28,18 @@ def trim(lis):
         del lis[-1]
     return lis
 
+def prefix(n):
+  return str(n)[:2] if len(str(n)) == 5 else str(n)[:1]
 
-URL = "http://112.186.226.178:4082"
+
+URL = "http://222.106.100.23:4082"
 
 
 class CONSTANT:
     __slots__ = ("PREFIX", "orgnum", "daynum", "thnum", "sbnum", "BASEURL", "SEARCHURL")
 
     async def refresh(self):
-        comci_resp = await AsyncRequest(f"{URL}/st")
+        comci_resp = await AsyncRequest(f"{URL}/st", "euc-kr")
 
         comcigan_html = BeautifulSoup(comci_resp, "lxml")
         script = comcigan_html.find_all("script")[1].contents[0]
@@ -120,11 +123,11 @@ async def refresh(self):
                 [
                     [
                         (
-                            subjects[int(str(x)[-2:])],
-                            long_subjects[int(str(x)[-2:])],
+                            subjects[int(prefix(x))],
+                            long_subjects[int(prefix(x))],
                             ""
-                            if int(str(x)[:-2]) >= len(teachers)
-                            else teachers[int(str(x)[:-2])],
+                            if int(str(x)[-2:]) >= len(teachers)
+                            else teachers[int(str(x)[-2:])],
                         )
                         for x in filter(lambda x: str(x)[:-2], trim(oneday[1:]))
                     ]

diff --git a/comcigan/school.py b/comcigan/school.py
@@ -22,8 +22,10 @@ def trim(lis):
         del lis[-1]
     return lis
 
+def prefix(n):
+  return str(n)[:2] if len(str(n)) == 5 else str(n)[:1]
 
-URL = "http://112.186.226.178:4082"
+URL = "http://222.106.100.23:4082"
 
 comci_resp = get(f"{URL}/st")
 comci_resp.encoding = "EUC-KR"
@@ -95,11 +97,11 @@ def refresh(self):
                 [
                     [
                         (
-                            subjects[int(str(x)[-2:])],
-                            long_subjects[int(str(x)[-2:])],
+                            subjects[int(prefix(x))],
+                            long_subjects[int(prefix(x))],
                             ""
-                            if int(str(x)[:-2]) >= len(teachers)
-                            else teachers[int(str(x)[:-2])],
+                            if int(str(x)[-2:]) >= len(teachers)
+                            else teachers[int(str(x)[-2:])],
                         )
                         for x in filter(lambda x: str(x)[:-2], trim(oneday[1:]))
                     ]