85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
|
"""utility models for use in other models"""
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.db import models
|
||
|
|
from django.utils import timezone
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
|
||
|
|
|
||
|
|
class Category(models.TextChoices):
|
||
|
|
"""what GLAMR are you"""
|
||
|
|
|
||
|
|
GALLERIES = "GAL", _("Galleries")
|
||
|
|
LIBRARIES = "LIB", _("Libraries")
|
||
|
|
ARCHIVES = "ARC", _("Archives")
|
||
|
|
MUSEUMS = "MUS", _("Museums")
|
||
|
|
RECORDS = "REC", _("Records")
|
||
|
|
DIGITAL_HUMANITIES = "DH", _("Digital Humanities")
|
||
|
|
GLAM = "GLAM", _("GLAMR")
|
||
|
|
|
||
|
|
|
||
|
|
class GroupType(models.TextChoices):
|
||
|
|
"""what GLAMR are you"""
|
||
|
|
|
||
|
|
DISCORD = "DISC", _("Discord server")
|
||
|
|
DISCOURSE = "DCRS", _("Discourse community")
|
||
|
|
EMAIL = "EML", _("email list")
|
||
|
|
GOOGLE = "GOOG", _("Google group")
|
||
|
|
KBIN = "KBIN", _("KBin server")
|
||
|
|
LEMMY = "LEMM", _("Lemmy server")
|
||
|
|
MASTODON = "MAS", _("Mastodon server")
|
||
|
|
REDDIT = "RED", _("subreddit")
|
||
|
|
SLACK = "SLAC", _("Slack channel")
|
||
|
|
ZULIP = "ZLIP", _("Zulip server")
|
||
|
|
OTHER = "OTHR", _("group")
|
||
|
|
|
||
|
|
|
||
|
|
class Announcement(models.Model):
|
||
|
|
"""an announcement on Mastodon"""
|
||
|
|
|
||
|
|
status = models.TextField()
|
||
|
|
summary = models.TextField(null=True)
|
||
|
|
queued = models.DateTimeField(default=timezone.now)
|
||
|
|
|
||
|
|
def save(self, *args, **kwargs):
|
||
|
|
if not self.queued:
|
||
|
|
self.queued = timezone.now()
|
||
|
|
super().save(*args, **kwargs)
|
||
|
|
|
||
|
|
def announce(self):
|
||
|
|
"""tell the world about it"""
|
||
|
|
|
||
|
|
key = settings.MASTODON_ACCESS_TOKEN
|
||
|
|
headers = {"Authorization": f"Bearer {key}"}
|
||
|
|
params = {"status": self.status}
|
||
|
|
if self.summary:
|
||
|
|
params["spoiler_text"] = self.summary
|
||
|
|
|
||
|
|
url = f"{settings.MASTODON_DOMAIN}/api/v1/statuses"
|
||
|
|
r = requests.post(url, data=params, headers=headers, timeout=(4, 13))
|
||
|
|
if r.status_code == 200:
|
||
|
|
self.delete()
|
||
|
|
|
||
|
|
|
||
|
|
class ContentWarning(models.Model):
|
||
|
|
"""content warnings"""
|
||
|
|
|
||
|
|
match_text = models.CharField(max_length=999, null=True)
|
||
|
|
display = models.CharField(max_length=999, null=True)
|
||
|
|
|
||
|
|
def is_in(self, text):
|
||
|
|
"""check some text and return a CW if needed"""
|
||
|
|
|
||
|
|
warning = None
|
||
|
|
if self.match_text in text.lower():
|
||
|
|
warning = self.display
|
||
|
|
return warning
|
||
|
|
|
||
|
|
|
||
|
|
class SiteMessage(models.Model):
|
||
|
|
"""A message to be displayed somewhere"""
|
||
|
|
|
||
|
|
message = models.TextField(max_length=999)
|