from django.db import models from django.utils.translation import ugettext_lazy as _ import django.contrib.postgres.fields as postgres from users.models import User from common.utils import GenerateDateUUIDMediaFilePath from django.conf import settings def sync_file_path(instance, filename): return GenerateDateUUIDMediaFilePath(instance, filename, settings.SYNC_FILE_PATH_ROOT) class SyncTask(models.Model): """A class that records information about douban data synchronization task.""" #-----------------------------------# # basic infos #-----------------------------------# user = models.ForeignKey( User, on_delete=models.CASCADE, related_name='user_%(class)ss') file = models.FileField(upload_to=sync_file_path, default='') #-----------------------------------# # options #-----------------------------------# # for the same book, if is already marked before sync, overwrite the previous mark or not overwrite = models.BooleanField(default=False) # sync book marks or not sync_book = models.BooleanField() # sync movie marks or not sync_movie = models.BooleanField() # sync music marks or not sync_music = models.BooleanField() # sync game marks or not sync_game = models.BooleanField() # default visibility of marks default_public = models.BooleanField() #-----------------------------------# # execution status #-----------------------------------# is_failed = models.BooleanField(default=False) # fail_reason = models.TextField(default='') is_finished = models.BooleanField(default=False) # how many items to synchronize total_items = models.PositiveIntegerField(default=0) # how many items are handled finished_items = models.PositiveIntegerField(default=0) # how many imtes have been synchronized successfully success_items = models.PositiveIntegerField(default=0) failed_urls = postgres.ArrayField( models.URLField(blank=True, default='', max_length=200), null=True, blank=True, default=list, ) break_point = models.CharField(max_length=100, default="", blank=True) started_time = models.DateTimeField(auto_now_add=True) ended_time = models.DateTimeField(auto_now=True) class Meta: """Meta definition for SyncTask.""" verbose_name = 'SyncTask' verbose_name_plural = 'SyncTasks' def __str__(self): """Unicode representation of SyncTask.""" return f'{self.id} {self.user} {self.file} {self.get_status_emoji()} {self.success_items}/{self.finished_items}/{self.total_items}' def get_status_emoji(self): return ("❌" if self.is_failed else "✔") if self.is_finished else "⚡" def get_duration(self): return self.ended_time - self.started_time def get_overwritten_items(self): if self.overwrite: return self.overwrite_books + self.overwrite_games + self.overwrite_movies + self.overwrite_music else: return 0 def get_progress(self): """ @return: return percentage """ if self.is_finished: return 100 else: if self.total_items > 0: return 100 * self.finished_items / self.total_items else: return 0 def get_breakpoint(self): if not self.break_point: return None, None progress = self.break_point.split('-') return progress[0], int(progress[1]) def set_breakpoint(self, sheet_name, row_index, save=False): self.break_point = f"{sheet_name}-{row_index}" if save: self.save(update_fields=['break_point']) def clear_breakpoint(self, save=False): self.break_point = "" if save: self.save(update_fields=['break_point'])