fix auto update with poor solution

phattd15 · Nov 15, 2021 · 7c52942 · 7c52942
1 parent d163d82
commit 7c52942
Show file tree

Hide file tree

Showing 5 changed files with 41 additions and 1 deletion.
diff --git a/base/migrations/0007_fetchdata.py b/base/migrations/0007_fetchdata.py
@@ -0,0 +1,20 @@
+# Generated by Django 3.2.9 on 2021-11-15 17:27
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('base', '0006_profile_history'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='FetchData',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('last_update', models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+    ]
diff --git a/base/migrations/__pycache__/0007_fetchdata.cpython-38.pyc b/base/migrations/__pycache__/0007_fetchdata.cpython-38.pyc
diff --git a/base/models.py b/base/models.py
@@ -56,3 +56,12 @@ class AuthQuery(models.Model):
     contest_id = models.IntegerField(default=1400)
     index = models.CharField(max_length=4)
     valid = models.BooleanField(default=False)
+
+
+"""
+TODO: Fix this with async / celery later
+"""
+
+
+class FetchData(models.Model):
+    last_update = models.DateTimeField(auto_now_add=True)
diff --git a/base/util.py b/base/util.py
@@ -2,7 +2,7 @@
 from django.contrib.auth.models import User
 import json
 from urllib.request import urlopen
-from .models import Problem
+from .models import Problem, FetchData
 from random import randint
 from datetime import datetime, timedelta
 from django.utils import timezone
@@ -31,8 +31,12 @@ def read_data(url):
 def fetch_problemset():
     URL = "https://codeforces.com/api/problemset.problems"
 
+    FetchData.objects.all().delete()
     Problem.objects.all().delete()
 
+    fd = FetchData()
+    fd.save()
+
     data = read_data(URL)
     problemset = data["problems"]
 

diff --git a/base/views.py b/base/views.py
@@ -1,3 +1,4 @@
+from datetime import time
 from django.shortcuts import redirect, render
 from .models import Profile, Problem, AuthQuery
 from django.contrib.auth.models import User
@@ -53,6 +54,12 @@ def challenge_list(request):
     if profile.in_progress:
         return redirect("challenge")
 
+    fd = FetchData.objects.all()
+
+    if fd[0].last_update + timedelta(hours=12) < timezone.now():
+        update_problemset()
+        fd[0].last_update = timezone.now()
+
     if request.method == "POST":
         contest_id = request.POST.get("contest_id")
         index = request.POST.get("index")