initial commit

login.py

@@ -0,0 +1,103 @@
+# **************************************************************************** #
+#                                                                              #
+#                                                         :::      ::::::::    #
+#    login.py                                           :+:      :+:    :+:    #
+#                                                     +:+ +:+         +:+      #
+#    By: tomoron <tomoron@student.42angouleme.fr>   +#+  +:+       +#+         #
+#                                                 +#+#+#+#+#+   +#+            #
+#    Created: 2024/11/25 16:22:08 by tomoron           #+#    #+#              #
+#    Updated: 2025/06/03 03:51:21 by tomoron          ###   ########.fr        #
+#                                                                              #
+# **************************************************************************** #
+
+import requests
+import time
+import subprocess
+import os
+import re
+import json
+import urllib.parse
+from getpass import getpass
+from bs4 import BeautifulSoup
+from collections import defaultdict
+
+
+class Intra42():
+    user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.6167.160 Safari/537.36"
+
+    def __init__(self):
+        self.s = requests.Session()
+        self.s.headers.update({"User-Agent": Intra42.user_agent})
+
+        #load refresh token
+        data = {}
+        self.load_session()
+        self.login_user()
+
+    def load_session():
+        if(not os.path.isfile('.cookies.json')):
+            return
+
+        with open(".cookies.json", 'r') as f:
+            data = json.loads(f.read())
+        
+        for domain, domain_cookies in data.items():
+            for cookie_name, cookie_value in domain_cookies.items():
+                new_cookie = requests.cookies.create_cookie(domain = domain, name = cookie_name, value = cookie_value)
+                self.s.cookies.set_cookie(new_cookie)
+        
+        
+    def login_user(self):
+        response = self.s.get("https://intra.42.fr/")
+
+        if(response.url == 'https://profile-v3.intra.42.fr/'):
+            return;
+        if(response.url == 'https://profile.intra.42.fr/'):
+            return;
+
+        page = BeautifulSoup(response.text, "html.parser")
+        form_html = page.find_all("form",id="kc-form-login")[0]
+        form_start = re.compile("^<form.*>").findall(str(form_html))[0]
+        url = form_start.split(' ')[1].split("\"")[1]
+    
+        username = input("login : ")
+        password = getpass("password : ")
+    
+        username = urllib.parse.quote(username)
+        password = urllib.parse.quote(password)
+
+        req_body = f'username={username}&password={password}&rememberMe=on&credentialId='
+        url = url.replace("amp;", "")
+        headers = { "Content-Type": "application/x-www-form-urlencoded"}
+
+        response = self.s.post(url, data=req_body ,headers=headers,allow_redirects=False)
+        if(response.status_code == 200):
+            print("invalid username or password")
+            exit()
+        print("sucessfuly logged in")
+
+        cookies = defaultdict(dict)
+        for cookie in self.s.cookies:
+            cookies[cookie.domain][cookie.name] = cookie.value
+
+        with open(".cookies.json", 'w') as f:
+            f.write(json.dumps(dict(cookies)))
+
+    def request_url(self, url, allow_redirects=False):
+        res = self.s.get(url, allow_redirects=allow_redirects)
+        if(res.status_code == 302):
+            login_intra()
+            return(self.request_url(url))
+        return(res)
+    
+    def get_intra_home(self):
+        response =  self.request_url("https://profile.intra.42.fr/")
+        return(response.text)
+    
+    def get_goals(self):
+        response = self.request_url("https://profile.intra.42.fr/users/me/goals?cursus=42cursus")
+        return(response.text)
+
+    def get_project_page(self, slug):
+        response = self.request_url(f"https://projects.intra.42.fr/projects/{slug}")
+        return(response.text)