Skip to content

Commit

Permalink
Refactoring Django code (#2277)
Browse files Browse the repository at this point in the history
  • Loading branch information
ibrahimjaved12 authored Aug 26, 2024
1 parent 877220a commit 4c2f662
Show file tree
Hide file tree
Showing 14 changed files with 47 additions and 63 deletions.
2 changes: 0 additions & 2 deletions content_sync/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
"""content_sync init file"""

default_app_config = "content_sync.apps.ContentSyncApp"
25 changes: 13 additions & 12 deletions content_sync/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from content_sync.models import ContentSyncState


@admin.register(ContentSyncState)
class ContentSyncStateAdmin(TimestampedModelAdmin):
"""ContentSyncState Admin"""

Expand All @@ -29,26 +30,26 @@ class ContentSyncStateAdmin(TimestampedModelAdmin):
def get_queryset(self, request): # noqa: ARG002
return self.model.objects.get_queryset().select_related("content__website")

@admin.display(
description="Content Title",
ordering="content__title",
)
def get_content_title(self, obj):
"""Returns the related WebsiteContent title""" # noqa: D401
return obj.content.title

get_content_title.short_description = "Content Title"
get_content_title.admin_order_field = "content__title"

@admin.display(
description="Content Text ID",
ordering="content__text_id",
)
def get_content_text_id(self, obj):
"""Returns the related WebsiteContent text ID""" # noqa: D401
return obj.content.text_id

get_content_text_id.short_description = "Content Text ID"
get_content_text_id.admin_order_field = "content__text_id"

@admin.display(
description="Website",
ordering="content__website__name",
)
def get_website_name(self, obj):
"""Returns the related Website name""" # noqa: D401
return obj.content.website.name

get_website_name.short_description = "Website"
get_website_name.admin_order_field = "content__website__name"


admin.site.register(ContentSyncState, ContentSyncStateAdmin)
25 changes: 13 additions & 12 deletions external_resources/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from external_resources.models import ExternalResourceState


@admin.register(ExternalResourceState)
class ExternalResourceStateAdmin(TimestampedModelAdmin):
"""ExternalResourceState Admin"""

Expand All @@ -31,26 +32,26 @@ def get_queryset(self, request): # noqa: ARG002
"""Return data along with the related WebsiteContent"""
return self.model.objects.get_queryset().select_related("content__website")

@admin.display(
description="Content Title",
ordering="content__title",
)
def get_content_title(self, obj):
"""Return the related WebsiteContent title"""
return obj.content.title

get_content_title.short_description = "Content Title"
get_content_title.admin_order_field = "content__title"

@admin.display(
description="Content Text ID",
ordering="content__text_id",
)
def get_content_text_id(self, obj):
"""Return the related WebsiteContent text ID"""
return obj.content.text_id

get_content_text_id.short_description = "Content Text ID"
get_content_text_id.admin_order_field = "content__text_id"

@admin.display(
description="Website",
ordering="content__website__name",
)
def get_website_name(self, obj):
"""Return the related Website name"""
return obj.content.website.name

get_website_name.short_description = "Website"
get_website_name.admin_order_field = "content__website__name"


admin.site.register(ExternalResourceState, ExternalResourceStateAdmin)
2 changes: 0 additions & 2 deletions gdrive_sync/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
"""gdrive_sync init file"""

default_app_config = "gdrive_sync.apps.GDriveSyncApp"
6 changes: 2 additions & 4 deletions gdrive_sync/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from gdrive_sync.models import DriveApiQueryTracker, DriveFile


@admin.register(DriveApiQueryTracker)
class DriveApiQueryTrackerAdmin(TimestampedModelAdmin):
"""DriveApiQueryTracker Admin"""

Expand All @@ -17,6 +18,7 @@ class DriveApiQueryTrackerAdmin(TimestampedModelAdmin):
)


@admin.register(DriveFile)
class DriveFileAdmin(TimestampedModelAdmin):
"""DriveFile Admin"""

Expand Down Expand Up @@ -45,7 +47,3 @@ class DriveFileAdmin(TimestampedModelAdmin):
"created_time",
"download_link",
]


admin.site.register(DriveFile, DriveFileAdmin)
admin.site.register(DriveApiQueryTracker, DriveApiQueryTrackerAdmin)
2 changes: 0 additions & 2 deletions main/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
"""Set the default AppConfig so we can validate settings"""

default_app_config = "main.apps.RootConfig"
7 changes: 3 additions & 4 deletions main/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,21 +15,20 @@
"""

from django.conf import settings
from django.conf.urls import include
from django.conf.urls.static import static
from django.contrib import admin
from django.contrib.auth.views import LogoutView
from django.urls import path, re_path
from django.urls import include, path, re_path

from main.views import public_index, restricted_index

urlpatterns = [
path("admin/", admin.site.urls),
path("robots.txt", include("robots.urls")),
path("", include("social_django.urls", namespace="social")),
re_path(r"^hijack/", include("hijack.urls", namespace="hijack")),
path("hijack/", include("hijack.urls", namespace="hijack")),
# Example view
re_path("^$", public_index, name="main-index"),
path("", public_index, name="main-index"),
path("login/", public_index, name="login"),
path("privacy-policy/", public_index, name="privacy-policy"),
re_path(r"^sites/.*$", restricted_index, name="sites"),
Expand Down
4 changes: 1 addition & 3 deletions users/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from users.models import User


@admin.register(User)
class UserAdmin(ContribUserAdmin, TimestampedModelAdmin):
"""Admin views for user"""

Expand Down Expand Up @@ -50,6 +51,3 @@ class UserAdmin(ContribUserAdmin, TimestampedModelAdmin):
search_fields = ("username", "name", "email")
ordering = ("email",)
readonly_fields = ("last_login",)


admin.site.register(User, UserAdmin)
2 changes: 0 additions & 2 deletions videos/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
"""videos init file"""

default_app_config = "videos.apps.VideoApp"
4 changes: 1 addition & 3 deletions videos/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def has_add_permission(self, request, obj=None): # noqa: ARG002
return False


@admin.register(Video)
class VideoAdmin(TimestampedModelAdmin):
"""Video Admin"""

Expand All @@ -48,6 +49,3 @@ class VideoAdmin(TimestampedModelAdmin):
)
list_filter = ("status",)
ordering = ("-updated_on",)


admin.site.register(Video, VideoAdmin)
10 changes: 5 additions & 5 deletions videos/urls.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
"""Urls for video"""

from django.urls import re_path
from django.urls import path, re_path

from videos.views import TranscodeJobView, TranscriptJobView, YoutubeTokensView

urlpatterns = [
re_path(
r"api/transcode-jobs/$",
path(
"api/transcode-jobs/",
TranscodeJobView.as_view(),
name="transcode_jobs",
),
re_path(
r"api/transcription-jobs/$",
path(
"api/transcription-jobs/",
TranscriptJobView.as_view(),
name="transcript_jobs",
),
Expand Down
2 changes: 0 additions & 2 deletions websites/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
"""websites init file"""

default_app_config = "websites.apps.WebsitesConfig"
15 changes: 7 additions & 8 deletions websites/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
from websites.models import Website, WebsiteContent, WebsiteStarter


@admin.register(Website)
class WebsiteAdmin(TimestampedModelAdmin, GuardedModelAdmin):
"""Website Admin"""

Expand All @@ -38,6 +39,7 @@ class WebsiteAdmin(TimestampedModelAdmin, GuardedModelAdmin):
ordering = ("-created_on",)


@admin.register(WebsiteContent)
class WebsiteContentAdmin(TimestampedModelAdmin, SafeDeleteAdmin):
"""WebsiteContent Admin"""

Expand Down Expand Up @@ -68,13 +70,14 @@ class WebsiteContentAdmin(TimestampedModelAdmin, SafeDeleteAdmin):
def get_queryset(self, request): # noqa: ARG002
return self.model.objects.get_queryset().select_related("website", "parent")

@admin.display(
description="Website",
ordering="website__title",
)
def get_website_title(self, obj):
"""Returns the related Website title""" # noqa: D401
return obj.website.title

get_website_title.short_description = "Website"
get_website_title.admin_order_field = "website__title"


class WebsiteStarterForm(forms.ModelForm):
"""Custom form for the WebsiteStarter model"""
Expand All @@ -100,6 +103,7 @@ def clean_config(self):
return config


@admin.register(WebsiteStarter)
class WebsiteStarterAdmin(TimestampedModelAdmin):
"""WebsiteStarter Admin"""

Expand All @@ -110,8 +114,3 @@ class WebsiteStarterAdmin(TimestampedModelAdmin):
list_display = ("id", "name", "status", "source", "commit")
list_filter = ("source",)
search_fields = ("name", "path")


admin.site.register(Website, WebsiteAdmin)
admin.site.register(WebsiteContent, WebsiteContentAdmin)
admin.site.register(WebsiteStarter, WebsiteStarterAdmin)
4 changes: 2 additions & 2 deletions websites/urls.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""websites URL Configuration"""

from django.urls import include, re_path
from django.urls import include, path
from rest_framework_extensions.routers import ExtendedSimpleRouter as SimpleRouter

from websites import views
Expand Down Expand Up @@ -33,5 +33,5 @@
)

urlpatterns = [
re_path(r"^api/", include(router.urls)),
path("api/", include(router.urls)),
]

0 comments on commit 4c2f662

Please sign in to comment.