Basic authentication views

fedireads/models.py

@@ -1,5 +1,6 @@
 ''' database schema for the whole dang thing '''
 from django.db import models
+from django.dispatch import receiver
 from django.contrib.auth.models import AbstractUser
 from django.contrib.postgres.fields import JSONField
 from Crypto.PublicKey import RSA
@@ -28,10 +29,27 @@ class User(AbstractUser):
 
         super().save(*args, **kwargs)
 
+@receiver(models.signals.post_save, sender=User)
+def execute_after_save(sender, instance, created, *args, **kwargs):
+    if not created:
+        return
+    shelves = [{
+        'name': 'To Read',
+        'type': 'to-read',
+    }, {
+        'name': 'Currently Reading',
+        'type': 'reading',
+    }, {
+        'name': 'Read',
+        'type': 'read',
+    }]
+
+    for shelf in shelves:
+        Shelf(name=shelf['name'], shelf_type=shelf['type'], user=instance, editable=False).save()
+
 
 class Message(models.Model):
     ''' any kind of user post, incl. reviews, replies, and status updates '''
-    id = models.AutoField(primary_key=True)
     author = models.ForeignKey('User', on_delete=models.PROTECT)
     name = models.CharField(max_length=255)
     content = JSONField(max_length=5000)
@@ -43,35 +61,52 @@ class Message(models.Model):
 
 
 class Review(Message):
-    id = models.AutoField(primary_key=True)
     book = models.ForeignKey('Book', on_delete=models.PROTECT)
     star_rating = models.IntegerField(default=0)
 
 
 class Shelf(models.Model):
-    id = models.AutoField(primary_key=True)
     name = models.CharField(max_length=100)
     user = models.ForeignKey('User', on_delete=models.PROTECT)
     editable = models.BooleanField(default=True)
-    books = models.ManyToManyField('Book', symmetrical=False)
+    shelf_type = models.CharField(default='custom', max_length=100)
+    books = models.ManyToManyField(
+        'Book',
+        symmetrical=False,
+        through='ShelfBook',
+        through_fields=('shelf', 'book')
+    )
     created_date = models.DateTimeField(auto_now_add=True)
     updated_date = models.DateTimeField(auto_now=True)
 
 
+class ShelfBook(models.Model):
+    # many to many join table for books and shelves
+    book = models.ForeignKey('Book', on_delete=models.PROTECT)
+    shelf = models.ForeignKey('Shelf', on_delete=models.PROTECT)
+    added_by = models.ForeignKey('User', blank=True, null=True, on_delete=models.PROTECT)
+    added_date = models.DateTimeField(auto_now_add=True)
+
+
 class Book(models.Model):
     ''' a non-canonical copy from open library '''
-    id = models.AutoField(primary_key=True)
     openlibary_key = models.CharField(max_length=255)
     data = JSONField()
     works = models.ManyToManyField('Work')
+    authors = models.ManyToManyField('Author')
     added_by = models.ForeignKey('User', on_delete=models.PROTECT, blank=True, null=True)
     added_date = models.DateTimeField(auto_now_add=True)
     updated_date = models.DateTimeField(auto_now=True)
 
 class Work(models.Model):
-    id = models.AutoField(primary_key=True)
     openlibary_key = models.CharField(max_length=255)
     data = JSONField()
-    added_by = models.ForeignKey('User', on_delete=models.PROTECT, blank=True, null=True)
     added_date = models.DateTimeField(auto_now_add=True)
     updated_date = models.DateTimeField(auto_now=True)
+
+class Author(models.Model):
+    openlibary_key = models.CharField(max_length=255)
+    data = JSONField()
+    added_date = models.DateTimeField(auto_now_add=True)
+    updated_date = models.DateTimeField(auto_now=True)
+