125 lines
3.5 KiB
Python
125 lines
3.5 KiB
Python
import threading
|
|
import time
|
|
|
|
from django.conf import settings
|
|
from django.contrib import admin
|
|
from django.contrib import messages
|
|
|
|
from django_admin_listfilter_dropdown.filters import DropdownFilter
|
|
import httpx
|
|
from mastodon import Mastodon
|
|
|
|
|
|
from . import models
|
|
|
|
|
|
def run_in_thread(fn):
|
|
def run(*k, **kw):
|
|
t = threading.Thread(target=fn, args=k, kwargs=kw)
|
|
t.start()
|
|
return t
|
|
return run
|
|
|
|
|
|
@admin.register(models.Person)
|
|
class AdminPerson(admin.ModelAdmin):
|
|
list_per_page = 50
|
|
list_display = (
|
|
'name',
|
|
'country',
|
|
'is_actor',
|
|
'is_director',
|
|
'is_woman',
|
|
)
|
|
search_fields = ('name',)
|
|
list_filter = ('is_woman', 'is_actor', 'is_director', 'country')
|
|
|
|
|
|
@admin.register(models.Movie)
|
|
class AdminMovie(admin.ModelAdmin):
|
|
list_per_page = 50
|
|
list_display = (
|
|
'name',
|
|
'original_name',
|
|
'year',
|
|
'duration',
|
|
'count')
|
|
search_fields = ('name', 'original_name')
|
|
filter_horizontal = ('directors',)
|
|
list_filter = (
|
|
'is_digital',
|
|
('directors__name', DropdownFilter),
|
|
('countries__name', DropdownFilter),
|
|
('year', DropdownFilter),
|
|
)
|
|
_is_new = False
|
|
actions = ['published']
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
self._is_new = obj.pk is None
|
|
super().save_model(request, obj, form, change)
|
|
return
|
|
|
|
def save_related(self, request, form, formsets, change):
|
|
obj = form.instance
|
|
super().save_related(request, form, formsets, change)
|
|
if self._is_new and obj.published:
|
|
self._public(obj)
|
|
return
|
|
|
|
def formfield_for_manytomany(self, db_field, request, **kwargs):
|
|
if db_field.name == 'directors':
|
|
kwargs['queryset'] = models.Person.objects.filter(is_director=True)
|
|
if db_field.name == 'actors':
|
|
kwargs['queryset'] = models.Person.objects.filter(is_actor=True)
|
|
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
|
|
|
def published(self, request, queryset):
|
|
obj = queryset[0]
|
|
self._public(obj)
|
|
self.message_user(request, 'Publicado correctamente', messages.SUCCESS)
|
|
published.short_description = 'Republicar en redes'
|
|
|
|
def _public_mastodon(self, message, cartel):
|
|
MT = {
|
|
'jpg': 'image/jpeg',
|
|
'png': 'image/png',
|
|
}
|
|
message += '\n\n#mauflix'
|
|
media_type = MT[cartel.url[-3:]]
|
|
|
|
try:
|
|
server = Mastodon(
|
|
access_token=settings.TOKEN_MASTODON,
|
|
api_base_url=settings.URL_MASTODON)
|
|
|
|
media = server.media_post(cartel.read(), media_type)
|
|
server.status_post(message, media_ids=media)
|
|
except Exception as e:
|
|
print(str(e))
|
|
return
|
|
|
|
def _public_telegram(self, message, cartel):
|
|
url = f'https://api.telegram.org/bot{settings.TOKEN_TELEGRAM}/sendPhoto'
|
|
url_cartel = settings.URL_CDN.format(cartel)
|
|
data = {'chat_id': settings.CHAT_ID, 'photo': url_cartel, 'caption': message}
|
|
result = httpx.post(url, data=data).json()
|
|
return
|
|
|
|
@run_in_thread
|
|
def _public(self, obj):
|
|
time.sleep(3)
|
|
message = f"""\"{obj.name}\"
|
|
Dir: {', '.join(obj.directors.values_list('name', flat=True))}
|
|
País: {', '.join(obj.countries.values_list('name', flat=True))}
|
|
Año: {obj.year}
|
|
"""
|
|
self._public_mastodon(message, obj.cartel)
|
|
self._public_telegram(message, obj.cartel.url)
|
|
return
|
|
|
|
|
|
admin.site.register(models.Gender)
|
|
admin.site.register(models.Country)
|
|
|