Hopiu/django-auditlog
1
0
Fork 0
mirror of https://github.com/jazzband/django-auditlog.git synced 2026-03-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[refactor]: Simplified truncate query class and remove redundant return statement

Browse source
This commit is contained in:
BinDruid 2024-10-20 22:31:15 +03:30
parent 9eb74cebbf
commit abeebf3e5e
1 changed files with 8 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
auditlog/management/commands/auditlogflush.py
View file

@ -66,21 +66,18 @@ class Command(BaseCommand):
entries = entries.filter(timestamp__date__lt=before)
count, _ = entries.delete()
self.stdout.write("Deleted %d objects." % count)
return
else:
database_vendor = connection.vendor
database_display_name = connection.display_name
table_name = LogEntry._meta.db_table
truncate_query = TruncateQuery(database_vendor, table_name)
if truncate_query.is_not_supported:
if not TruncateQuery.support_truncate_statement(database_vendor):
self.stdout.write(
"Database %s does not support truncate statement."
% database_display_name
)
return
with connection.cursor() as cursor:
query = truncate_query.to_sql()
query = TruncateQuery.to_sql(table_name)
cursor.execute(query)
self.stdout.write("Truncated log entry table.")
@ -88,13 +85,10 @@ class Command(BaseCommand):
class TruncateQuery:
SUPPORTED_VENDORS = ("postgresql", "mysql", "sqlite", "oracle", "microsoft")
def __init__(self, database_vendor: str, table_name: str):
self.database_vendor = database_vendor
self.table_name = table_name
@classmethod
def support_truncate_statement(cls, database_vendor) -> bool:
return database_vendor in cls.SUPPORTED_VENDORS
@property
def is_not_supported(self):
return self.database_vendor not in self.SUPPORTED_VENDORS
def to_sql(self):
return f"TRUNCATE TABLE {self.table_name};"
@staticmethod
def to_sql(table_name) -> str:
return f"TRUNCATE TABLE {table_name};"