Feat issue 1

sangmyung_univ_auth/__init__.py

@@ -0,0 +1 @@
+from .auth import auth

sangmyung_univ_auth/auth.py

@@ -0,0 +1,13 @@
+from sangmyung_univ_auth.authenticate import session_request, get_userinfo
+from sangmyung_univ_auth.response import AuthResponse, _auth_failed, _success, _unknown_issue
+
+
+def auth(username: str, password: str) -> AuthResponse:
+    try:
+        session = session_request(username, password)
+        if not session:
+            return _auth_failed()
+        userinfo: dict = get_userinfo(session)
+        return _success(body=userinfo)
+    except Exception:
+        return _unknown_issue()

sangmyung_univ_auth/authenticate.py

@@ -0,0 +1,23 @@
+import requests
+from requests import Session
+from bs4 import BeautifulSoup as bs
+
+
+def session_request(username: str, password: str):
+    with requests.session() as s:
+        user_info = {"username": username, "password": password}
+        request = s.post("https://ecampus.smu.ac.kr/login/index.php", data=user_info)
+        if request.url == "https://ecampus.smu.ac.kr/":
+            return s
+        return
+
+
+def get_userinfo(session: Session) -> dict:
+    request = session.get("https://ecampus.smu.ac.kr/user/user_edit.php")
+    source = request.text
+    soup = bs(source, "html.parser")
+    return {
+        "name": soup.find('input', id='id_firstname').get('value'),
+        "department": soup.find('input', id='id_department').get('value'),
+        "email": soup.find('input', id='id_email').get('value')
+    }

sangmyung_univ_auth/response.py

@@ -0,0 +1,36 @@
+from collections import namedtuple
+
+AuthResponse = namedtuple(
+    'AuthResponse', [
+        'is_auth',      # bool or None: 인증 여부
+        'code',         # str: 반환 코드
+        'body'          # Any: 메타데이터
+    ]
+)
+
+
+def _success(body=None):
+    return AuthResponse(
+        is_auth=True,
+        code='success',
+        body=body or {}
+    )
+
+
+def _auth_failed():
+    return AuthResponse(
+        is_auth=False,
+        code='auth_failed',
+        body={"message": "아이디 및 비밀번호가 일치하지 않습니다."}
+    )
+
+
+def _unknown_issue():
+    return AuthResponse(
+        is_auth=None,
+        code='unknown_issue',
+        body={
+            'message': '모듈이 예상한 포맷과 다릅니다. 관리자에게 문의해주세요! '
+                       '[https://github.com/hyunmin0317/sangmyung-univ-auth/issues]'
+        }
+    )

tests/test_auth.py

@@ -0,0 +1,19 @@
+import os
+import unittest
+from sangmyung_univ_auth import auth
+
+
+class MyTestCase(unittest.TestCase):
+    def setUp(self):
+        self.username = os.getenv('USERNAME')
+        self.password = os.getenv('PASSWORD')
+
+    def test_authenticator(self):
+        res = auth(username=self.username, password=self.password)
+        self.assertTrue(res.is_auth)
+        res = auth(username='username', password='password')
+        self.assertFalse(res.is_auth)
+
+
+if __name__ == '__main__':
+    unittest.main()