27 lines
841 B
Python
27 lines
841 B
Python
|
from django.conf import settings
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from loguru import logger
|
||
|
|
from tqdm import tqdm
|
||
|
|
|
||
|
|
from users.models import Preference, User
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = "Manage import tasks"
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument(
|
||
|
|
"--reset",
|
||
|
|
action="store_true",
|
||
|
|
)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
if options["reset"]:
|
||
|
|
count = 0
|
||
|
|
for user in tqdm(User.objects.all()):
|
||
|
|
if user.preference.import_status.get("douban_pending"):
|
||
|
|
user.preference.import_status["douban_pending"] = False
|
||
|
|
user.preference.save(update_fields=["import_status"])
|
||
|
|
count += 1
|
||
|
|
self.stdout.write(self.style.SUCCESS(f"{count} users reset"))
|