mirror of
https://github.com/th3r00t/pyShelf.git
synced 2026-04-28 01:59:35 -04:00
153 lines
4.6 KiB
Python
Executable File
153 lines
4.6 KiB
Python
Executable File
from django.contrib.postgres.search import SearchVector
|
|
from django.db import models
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import AbstractUser, User
|
|
from django.contrib.auth import get_user_model
|
|
|
|
# Create your models here.
|
|
|
|
|
|
class Books(models.Model):
|
|
"""
|
|
pyShelfs Book Database class
|
|
:param title: Book title
|
|
:param author: Author
|
|
:param categories: Categories <-- Not implemented
|
|
:param cover: Cover image BinaryField
|
|
:param pages: # of pages <-- Not implemented
|
|
:param progress: Reader percentage <-- Not implented
|
|
:param file_name: Path to book
|
|
"""
|
|
|
|
class Meta:
|
|
db_table = "books"
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
title = models.TextField(max_length=None)
|
|
author = models.CharField(max_length=255, null=True)
|
|
categories = models.TextField(max_length=None, null=True)
|
|
cover = models.BinaryField(null=True, editable=True)
|
|
pages = models.IntegerField(null=True)
|
|
progress = models.IntegerField(null=True)
|
|
file_name = models.TextField(max_length=None, null=False)
|
|
date = models.DateTimeField(auto_now_add=True)
|
|
description = models.TextField(null=True)
|
|
identifier = models.CharField(max_length=255, null=True)
|
|
publisher = models.TextField(max_length=None, null=True)
|
|
date = models.DateField(null=True)
|
|
rights = models.CharField(max_length=255, null=True)
|
|
tags = models.TextField(max_length=None, null=True)
|
|
|
|
def generic_search(self, query):
|
|
try:
|
|
results = Books.objects.annotate(
|
|
search=SearchVector("title", "file_name", "author", "tags"),
|
|
).filter(search=query)
|
|
except Exception as e:
|
|
raise
|
|
return results
|
|
|
|
|
|
class Collections(models.Model):
|
|
class Meta:
|
|
db_table = "collections"
|
|
|
|
def __str__(self):
|
|
return self.collection
|
|
|
|
collection = models.CharField(max_length=255)
|
|
book_id = models.ForeignKey(Books, on_delete=models.PROTECT)
|
|
|
|
def get_absolute_url(self):
|
|
"""Returns the url to access a particular instance of MyModelName."""
|
|
return reverse("model-detail-view", args=[str(self.id)])
|
|
|
|
def generic_search(self, query):
|
|
try:
|
|
results = Books.objects.annotate(search=SearchVector("collection"),).filter(
|
|
search=query
|
|
)
|
|
except Exception as e:
|
|
raise
|
|
return results
|
|
|
|
|
|
class Navigation(models.Model):
|
|
"""
|
|
pyShelfs Navigation Database class
|
|
:param title: Link Text
|
|
:param link: Link link :)
|
|
:param category: Where in the nav tree do I belong
|
|
:param parent_id: This link is a sub link of link with id of me
|
|
:param alt: Alternate text of link
|
|
:param type: Web link, or Socket link which will be expected to act on \
|
|
the link, and the action defined in socket
|
|
:param socket: if a Socket link define socket here
|
|
"""
|
|
|
|
class Meta:
|
|
db_table = "navigation"
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
title = models.CharField(max_length=255)
|
|
link = models.CharField(max_length=255, null=True)
|
|
category = models.CharField(max_length=255, null=True)
|
|
parent_id = models.IntegerField(null=True, editable=True)
|
|
alt = models.CharField(max_length=255, null=True)
|
|
type = models.IntegerField(null=True)
|
|
socket = models.CharField(max_length=255, null=False)
|
|
|
|
def generic_search(self, query):
|
|
try:
|
|
results = Navigation.objects.annotate(
|
|
search=SearchVector("title", "parent_id", "category"),
|
|
).filter(search=query)
|
|
except Exception as e:
|
|
raise
|
|
return results
|
|
|
|
|
|
class User(AbstractUser):
|
|
facebook = models.CharField(max_length=255, null=True)
|
|
twitter = models.CharField(max_length=255, null=True)
|
|
ulvl = models.IntegerField(default=1)
|
|
sponsorid = models.IntegerField(null=True)
|
|
matrixid = models.CharField(max_length=255, null=True)
|
|
|
|
def __str__(self):
|
|
return self.username
|
|
|
|
|
|
class Favorites(models.Model):
|
|
"""
|
|
Favorites Database class
|
|
:param book: book foreign key
|
|
:param user: user foreign key
|
|
"""
|
|
|
|
class Meta:
|
|
db_table = "favorites"
|
|
|
|
def __str__(self):
|
|
return self.book
|
|
|
|
book = models.ForeignKey(Books, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(
|
|
User,
|
|
on_delete=models.CASCADE
|
|
)
|
|
def generic_search(self, query):
|
|
try:
|
|
results = Favorites.objects.annotate(search=SearchVector("user"),).filter(
|
|
search=query
|
|
)
|
|
except Exception as e:
|
|
raise
|
|
return results
|
|
|
|
|