__all__ = ['Author', 'Book', 'Extension', 'Genre', 'Language',
'AuthorBook', 'BookGenre',
- 'init_db', 'insert_name', 'update_counters',
+ 'init_db', 'insert_name', 'insert_author', 'update_counters',
]
import os
class Author(SQLObject):
- name = UnicodeCol(unique=True)
- count = IntCol()
+ surname = UnicodeCol(notNull=True)
+ name = UnicodeCol(notNull=True)
+ misc_name = UnicodeCol(notNull=True)
+ count = IntCol(notNull=True)
books = RelatedJoin('Book', otherColumn='book_id',
intermediateTable='author_book',
createRelatedTable=False)
+ full_name_idx = DatabaseIndex(surname, name, misc_name, unique=True)
count_idx = DatabaseIndex(count)
genres = RelatedJoin('Genre',
intermediateTable='book_genre',
createRelatedTable=False)
- title = UnicodeCol()
- series = UnicodeCol()
+ title = UnicodeCol(notNull=True)
+ series = UnicodeCol(notNull=True)
ser_no = IntCol()
- archive = StringCol()
- file = StringCol()
- size = IntCol()
- lib_id = StringCol()
- deleted = BoolCol()
- extension = ForeignKey('Extension')
- date = DateCol()
+ archive = StringCol(notNull=True)
+ file = StringCol(notNull=True)
+ size = IntCol(notNull=True)
+ lib_id = StringCol(notNull=True)
+ deleted = BoolCol(notNull=True)
+ extension = ForeignKey('Extension', notNull=True)
+ date = DateCol(notNull=True)
language = ForeignKey('Language')
title_idx = DatabaseIndex(title)
series_idx = DatabaseIndex(series)
class Extension(SQLObject):
- name = StringCol(unique=True)
- count = IntCol()
+ name = StringCol(notNull=True, unique=True)
+ count = IntCol(notNull=True)
count_idx = DatabaseIndex(count)
class Genre(SQLObject):
- name = StringCol(unique=True)
- title = UnicodeCol()
- count = IntCol()
+ name = StringCol(notNull=True, unique=True)
+ title = UnicodeCol(notNull=True)
+ count = IntCol(notNull=True)
books = RelatedJoin('Book', otherColumn='book_id',
intermediateTable='book_genre',
createRelatedTable=False)
class Language(SQLObject):
- name = StringCol(unique=True)
- count = IntCol()
+ name = StringCol(notNull=True, unique=True)
+ count = IntCol(notNull=True)
count_idx = DatabaseIndex(count)
return table(name=name, count=0, **kw)
+def insert_author(surname, name, misc_name):
+ try:
+ return Author.full_name_idx.get(
+ surname=surname, name=name, misc_name=misc_name)
+ except SQLObjectNotFound:
+ return Author(surname=surname, name=name, misc_name=misc_name, count=0)
+
+
def update_counters():
for author in Author.select():
author.count = len(author.books)