from django.db import models

# Create your models here.


class Book (models.Model):
    ID = models.AutoField(primary_key=True)
    name = models.TextField(max_length=128)
    description = models.TextField(max_length=256, null=True)
    category = models.TextField(max_length=32)
    serial_number = models.TextField(max_length=128)
    writer = models.TextField(max_length=128)
    status = models.TextField(max_length=32)
    borrower = models.TextField(max_length=128, null=True)
    borrower_phone = models.IntegerField(null=True)
    borrow_time = models.DateField(null=True)

    def create_new(self):
        Book.objects.create(name=self.name, category=self.category, status='在库', description=self.description, serial_number=self.serial_number, writer=self.writer)

    def update(self):
        Book.objects.filter(ID=self.ID).update(name=self.name, category=self.category, status=self.status,
                                               description=self.description, borrower=self.borrower,
                                               borrower_phone=self.borrower_phone, borrow_time=self.borrow_time, serial_number=self.serial_number, writer=self.writer)
    def getList(self):
        return Book.objects.all().order_by('serial_number')

    def search_by_ID(self):
        return Book.objects.filter(ID=self.ID)

    def search_filter(self, rules):
        return Book.objects.filter(rules)


class User (models.Model):
    ID = models.AutoField(primary_key=True)
    name = models.TextField(max_length=32)
    phone = models.IntegerField()

    def create_new(self):
        User.objects.create(ID=self.ID, name=self.name, phone=self.phone)

    def update(self):
        User.objects.filter(ID=self).update(name=self.name, phone=self.phone)

    def search_by_name(self):
        try:
            return User.objects.filter(name=self.name)
        except:
            return None