Allow profile disable of lastfm imports

This commit is contained in:
2023-04-06 13:37:10 -04:00
parent 59e29d858a
commit 8042c726b0
3 changed files with 22 additions and 1 deletions

View File

@ -0,0 +1,18 @@
# Generated by Django 4.1.7 on 2023-04-06 17:36
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
("profiles", "0007_alter_userprofile_timezone"),
]
operations = [
migrations.AddField(
model_name="userprofile",
name="lastfm_auto_import",
field=models.BooleanField(default=False),
),
]

View File

@ -23,6 +23,7 @@ class UserProfile(TimeStampedModel):
) )
lastfm_username = models.CharField(max_length=255, **BNULL) lastfm_username = models.CharField(max_length=255, **BNULL)
lastfm_password = EncryptedField(**BNULL) lastfm_password = EncryptedField(**BNULL)
lastfm_auto_import = models.BooleanField(default=False)
def __str__(self): def __str__(self):
return f"User profile for {self.user}" return f"User profile for {self.user}"

View File

@ -197,7 +197,9 @@ def import_lastfm_for_all_users():
"""Grab a list of all users with LastFM enabled and kickoff imports for them""" """Grab a list of all users with LastFM enabled and kickoff imports for them"""
LastFmImport = apps.get_model("scrobbles", "LastFMImport") LastFmImport = apps.get_model("scrobbles", "LastFMImport")
lastfm_enabled_user_ids = UserProfile.objects.filter( lastfm_enabled_user_ids = UserProfile.objects.filter(
lastfm_username__isnull=False, lastfm_password__isnull=False lastfm_username__isnull=False,
lastfm_password__isnull=False,
lastfm_auto_import=True,
).values_list("user_id", flat=True) ).values_list("user_id", flat=True)
for user_id in lastfm_enabled_user_ids: for user_id in lastfm_enabled_user_ids: