dev: fix imports

This commit is contained in:
pablohashescobar 2024-03-27 15:57:05 +05:30
parent 6cc29dca52
commit 6a6d0d881c
4 changed files with 15 additions and 30 deletions

View File

@ -78,7 +78,7 @@ class IssueSerializer(BaseSerializer):
parsed_str = html.tostring(parsed, encoding="unicode") parsed_str = html.tostring(parsed, encoding="unicode")
data["description_html"] = parsed_str data["description_html"] = parsed_str
except Exception as e: except Exception:
raise serializers.ValidationError("Invalid HTML passed") raise serializers.ValidationError("Invalid HTML passed")
# Validate assignees are from project # Validate assignees are from project
@ -349,7 +349,7 @@ class IssueCommentSerializer(BaseSerializer):
parsed_str = html.tostring(parsed, encoding="unicode") parsed_str = html.tostring(parsed, encoding="unicode")
data["comment_html"] = parsed_str data["comment_html"] = parsed_str
except Exception as e: except Exception:
raise serializers.ValidationError("Invalid HTML passed") raise serializers.ValidationError("Invalid HTML passed")
return data return data

View File

@ -1,13 +1,13 @@
# Python imports # Python imports
from datetime import timedelta from datetime import timedelta
# Django imports
from django.utils import timezone
from django.db.models import Q
# Third party imports # Third party imports
from celery import shared_task from celery import shared_task
# Django imports
from django.db.models import Q
from django.utils import timezone
# Module imports # Module imports
from plane.db.models import FileAsset from plane.db.models import FileAsset
@ -29,25 +29,8 @@ def delete_file_asset():
@shared_task @shared_task
def file_asset_size(slug, email, members, issue_count, cycle_count, module_count): def file_asset_size():
asset_size = [] asset_size = []
# s3_client = boto3.client('s3')
assets_to_update = []
# for asset in FileAsset.objects.filter(size__isnull=True):
# try:
# key = f"{workspace_id}/{asset_key}"
# response = s3_client.head_object(Bucket=settings.AWS_STORAGE_BUCKET_NAME, Key=key)
# size = response['ContentLength']
# asset.size = size
# assets_to_update.append(asset)
# except Exception as e:
# # Handle exceptions such as S3 object not found
# print(f"Error updating asset size for {asset.asset.key}: {e}")
# # Bulk update only objects that need updating
# FileAsset.objects.bulk_update(assets_to_update, ["size"], batch_size=50)
for asset in FileAsset.objects.filter(size__isnull=True): for asset in FileAsset.objects.filter(size__isnull=True):
asset.size = asset.asset.size asset.size = asset.asset.size
asset_size.append(asset) asset_size.append(asset)

View File

@ -1,23 +1,21 @@
# Python imports # Python imports
import getpass
# Django imports # Django imports
from django.core.management import BaseCommand from django.core.management import BaseCommand
# Module imports # Module imports
from plane.db.models import User
class Command(BaseCommand): class Command(BaseCommand):
help = "Check the file asset size of the file" help = "Check the file asset size of the file"
def handle(self, *args, **options): def handle(self, *args, **options):
from plane.bgtasks.file_asset_task import file_asset_size from plane.bgtasks.file_asset_task import file_asset_size
# Start the queueing
file_asset_size.delay() file_asset_size.delay()
self.stdout.write( self.stdout.write(
self.style.SUCCESS(f"File asset size pushed to queue") self.style.SUCCESS("File asset size pushed to queue")
) )

View File

@ -1,5 +1,9 @@
# Python imports
import uuid
# Django imports # Django imports
from django.conf import settings from django.conf import settings
from django.core.exceptions import ValidationError
from django.core.validators import MaxValueValidator, MinValueValidator from django.core.validators import MaxValueValidator, MinValueValidator
from django.db import models from django.db import models
from django.db.models.signals import post_save from django.db.models.signals import post_save
@ -94,8 +98,8 @@ class IssueManager(models.Manager):
def get_upload_path(instance, filename): def get_upload_path(instance, filename):
if instance.workspace_id is not None: if instance.workspace_id is not None:
return f"{instance.workspace.id}/{uuid4().hex}" return f"{instance.workspace.id}/{uuid.uuid4().hex}"
return f"user-{uuid4().hex}" return f"user-{uuid.uuid4().hex}"
def file_size(value): def file_size(value):