from django.contrib import admin
from django.utils.html import format_html
from import_export.admin import ImportExportModelAdmin
from .models import streamingData
from .utils import full_testing
from urllib.parse import urlsplit, urlunsplit



@admin.action(description="check the URLs")
def check(modeladmin, request, queryset):
    for query in queryset:
        website_url = query.url
        status=full_testing(website_url)[0]
        result= full_testing(website_url)[1]
        print(str(query.url)+": "+str(status)+"  ->  "+str(result))
        query.status = status
        query.result = result
        query.save(update_fields=["status","result"])
        

#    modeladmin.message_user(
#        request,
#        ngettext(
#            "%d episode was successfully marked as seen.",
#            "%d episodes were successfully marked as seen.",
#            updated,
#        )
#        % updated,
#        messages.SUCCESS,
#        )

@admin.action(description="cleaning the url")
def urlcleaning(modeladmin, request, queryset):
    for query in queryset:
        # Split the URL into components
        parsed = urlsplit(query.url)
        
        # Rebuild using only scheme and netloc, adding a trailing slash at the end
        base_url = urlunsplit((parsed.scheme, parsed.netloc, "/", "", ""))
        query.url = base_url
        query.save(update_fields=["url"])
        print("Cleaning :"+str(base_url))
        
    return 0
@admin.register(streamingData)
class streamingAdmin(ImportExportModelAdmin):
    list_display = ['clickable_links','result','status']
    fields = ['url']
    actions = [check,urlcleaning]

    def clickable_links(self, obj):
        return format_html('<a href="{}">{}</a>', obj.url,obj.url)