Patch username changing

phattd15 · Dec 25, 2021 · 8d8bd08 · 8d8bd08
1 parent 378be14
commit 8d8bd08
Show file tree

Hide file tree

Showing 13 changed files with 36 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,7 +1,8 @@
 *.env
 /arugo/.env
 /db.sqlite3
-/base/__pycache__
-/arugo/__pycache__
+/base/__pycache__/
+/arugo/__pycache__/
 /base/__init__.py
 /arugo/__init__.py
+/base/migrations/__pycache__/
diff --git a/base/migrations/__pycache__/0001_initial.cpython-38.pyc b/base/migrations/__pycache__/0001_initial.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0002_alter_profile_deadline.cpython-38.pyc b/base/migrations/__pycache__/0002_alter_profile_deadline.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0003_auto_20211111_0801.cpython-38.pyc b/base/migrations/__pycache__/0003_auto_20211111_0801.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0004_authquery.cpython-38.pyc b/base/migrations/__pycache__/0004_authquery.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0005_authquery_password.cpython-38.pyc b/base/migrations/__pycache__/0005_authquery_password.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0006_profile_history.cpython-38.pyc b/base/migrations/__pycache__/0006_profile_history.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0007_fetchdata.cpython-38.pyc b/base/migrations/__pycache__/0007_fetchdata.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0008_profile_msg.cpython-38.pyc b/base/migrations/__pycache__/0008_profile_msg.cpython-38.pyc
diff --git a/base/migrations/__pycache__/0009_auto_20211207_1021.cpython-38.pyc b/base/migrations/__pycache__/0009_auto_20211207_1021.cpython-38.pyc
diff --git a/base/migrations/__pycache__/__init__.cpython-38.pyc b/base/migrations/__pycache__/__init__.cpython-38.pyc
diff --git a/base/util.py b/base/util.py
@@ -10,7 +10,7 @@
 from io import StringIO
 import numpy as np
 import math
-
+import requests
 
 def read_data(url):
     try:
@@ -542,3 +542,32 @@ def validate_auth_query(query):
 def discard_challenge(profile):
     profile.in_progress = False
     profile.save()
+
+def update_username(username):
+    url = 'https://codeforces.com/profile/' + username
+    res = requests.get(url)
+    res_url = res.url
+    slash_pos = 0
+    for i in range(len(res_url)):
+        if res_url[i] == '/':
+            slash_pos = i
+    nxt_username = res_url[(slash_pos + 1):]
+    # return nxt_username
+
+    if nxt_username != username:
+        user = User.objects.get(username=username)
+        profile = Profile.objects.get(handle=username)
+        queries = AuthQuery.objects.filter(handle=username)
+
+        if len(queries):
+            queries.delete()
+
+        profile.handle = nxt_username
+        profile.save()
+
+        user.username = nxt_username
+        user.save()
+
+        print("updated to " + nxt_username)
+
+    return nxt_username
diff --git a/base/views.py b/base/views.py
@@ -19,6 +19,7 @@ def index(request):
     context["user_count"] = len(User.objects.all())
 
     if request.user.is_authenticated:
+        print(user.username)
         profile = Profile.objects.get(user=user)
 
         if request.method == "POST" and user.check_password(
@@ -109,6 +110,8 @@ def login_view(request):
         user = authenticate(request, username=username, password=password)
 
         if user is not None:
+            username = update_username(username)
+            user = authenticate(request, username=username, password=password)
             login(request, user)
             response = redirect("home-page")
             return response