diff --git a/cps/db.py b/cps/db.py index bfceabe9..52c350da 100755 --- a/cps/db.py +++ b/cps/db.py @@ -49,17 +49,24 @@ books_languages_link = Table('books_languages_link', Base.metadata, Column('lang_code', Integer, ForeignKey('languages.id'), primary_key=True) ) -#cc = conn.execute("SELECT id FROM custom_columns") -#cc_ids = [] -#books_custom_column_links = {} -#for row in cc: -# cc_link=Table('books_custom_column_' + str(row.id) + '_link', Base.metadata, -# Column('book', Integer, ForeignKey('books.id'), primary_key=True), -# Column('custom_column', Integer, ForeignKey('custom_column_' + str(row.id) + '.id'), primary_key=True) -# ) -# books_custom_column_links[row.id] = cc_link -# cc_ids.append(row.id) - +cc = conn.execute("SELECT id FROM custom_columns") +cc_ids = [] + +books_custom_column_links = {} +for row in cc: + books_custom_column_links[row.id] = Table('books_custom_column_' + str(row.id) + '_link', Base.metadata, + Column('book', Integer, ForeignKey('books.id'), primary_key=True), + Column('value', Integer, ForeignKey('custom_column_' + str(row.id) + '.id'), primary_key=True) + ) + #books_custom_column_links[row.id]= + cc_ids.append(row.id) + +cc_classes = {} +for id in cc_ids: + ccdict={'__tablename__':'custom_column_' + str(id), + 'id':Column(Integer, primary_key=True), + 'value':Column(String)} + cc_classes[id] = type('Custom_Column_' + str(id), (Base,), ccdict) class Comments(Base): __tablename__ = 'comments' @@ -163,7 +170,7 @@ class Data(Base): class Books(Base): __tablename__ = 'books' - id = Column(Integer,primary_key=True) + id = Column(Integer, primary_key=True) title = Column(String) sort = Column(String) author_sort = Column(String) @@ -181,17 +188,7 @@ class Books(Base): series = relationship('Series', secondary=books_series_link, backref='books') ratings = relationship('Ratings', secondary=books_ratings_link, backref='books') languages = relationship('Languages', secondary=books_languages_link, backref='books') - #custom_columns = {} - #for id in cc_ids: - # print id - # custom_columns[id] = relationship(cc_classes[id], secondary=books_custom_column_links[id], backref='books') - #custom_columns[1] = relationship(cc_classes[1], secondary=books_custom_column_links[1], backref='books') - #custom_columns[2] = relationship(cc_classes[2], secondary=books_custom_column_links[2], backref='books') - #custom_columns[3] = relationship(cc_classes[3], secondary=books_custom_column_links[3], backref='books') - - - def __init__(self, title, sort, author_sort, timestamp, pubdate, series_index, last_modified, path, has_cover, authors, tags): self.title = title self.sort = sort @@ -205,7 +202,9 @@ class Books(Base): def __repr__(self): return u"".format(self.title, self.sort, self.author_sort, self.timestamp, self.pubdate, self.series_index, self.last_modified ,self.path, self.has_cover) - +for id in cc_ids: + setattr(Books, 'custom_column_' + str(id), relationship(cc_classes[id], secondary=books_custom_column_links[id], backref='books')) + class Custom_Columns(Base): __tablename__ = 'custom_columns' @@ -217,30 +216,9 @@ class Custom_Columns(Base): editable = Column(Boolean) display = Column(String) is_multiple = Column(Boolean) - normalized = Column(Boolean) - -#class Custom_Column(object): -# def __init__(self, value): -# self.value = value - -#def get_cc_table(id): -# custom_column = Custom_Column -# table_name = 'custom_column_' + str(id) -# table_object = Table(table_name, Base.metadata, -# Column('id', Integer, primary_key=True, autoincrement=True), -# Column('value', String) -# ) -# clear_mappers() -# mapper(custom_column, table_object, properties={'books' + str(id): relationship(Books, secondary=books_custom_column_links[id] , backref='custom_column_' + str(id))}) -# return custom_column - -#cc_classes={} -#for id in cc_ids: -# cc_classes[id] = get_cc_table(id) -#print cc_classes - + normalized = Column(Boolean) -Base.metadata.create_all(engine) +#Base.metadata.create_all(engine) Session = sessionmaker() Session.configure(bind=engine) session = Session() diff --git a/cps/helper.py b/cps/helper.py index 629c8109..4bd9d952 100755 --- a/cps/helper.py +++ b/cps/helper.py @@ -200,8 +200,8 @@ def update_dir_stucture(book_id): book.path = new_authordir + "/" + book.path.split("/")[1] db.session.commit() -def get_custom_columns(id): - cc = db.session.query(db.Custom_Columns).all() - for c in cc: - print c.name +#def get_custom_columns(id): +# cc = db.session.query(db.Custom_Columns).all() +# for c in cc: +# print c.name diff --git a/cps/templates/detail.html b/cps/templates/detail.html index bcaad48a..5cf9d5e2 100644 --- a/cps/templates/detail.html +++ b/cps/templates/detail.html @@ -58,6 +58,23 @@

{% endif %} + + {% if cc|length > 0 %} +

+

+ {% for c in cc %} + {% if entry['custom_column_' ~ c.id]|length > 0 %} + {{ c.name }}: + {% for column in entry['custom_column_' ~ c.id] %} + {{ column.value }} + {% endfor %} +
+ {% endif %} + {% endfor %} +
+

+ {% endif %} + {% if entry.comments|length > 0 %}

Description:

diff --git a/cps/templates/edit_book.html b/cps/templates/edit_book.html index b4ec3661..a3746e4e 100644 --- a/cps/templates/edit_book.html +++ b/cps/templates/edit_book.html @@ -43,6 +43,25 @@ + {% if cc|length > 0 %} + {% for c in cc %} +
+ + 0 %} + {% for column in book['custom_column_' ~ c.id] %} + value="{{ column.value }} {% if not loop.last %}, {% endif %} + {% endfor %} + " + {% endif %} + > +
+ {% endfor %} +
+

+ {% endif %} + +