support int custom fields

cps/db.py

@@ -11,7 +11,7 @@ from ub import config
 import ub
 
 session = None
-cc_exceptions = ['datetime', 'int', 'comments', 'float', 'composite', 'series']
+cc_exceptions = ['datetime', 'comments', 'float', 'composite', 'series']
 cc_classes = None
 engine = None
 
@@ -335,11 +335,18 @@ def setup_db():
                                                                  primary_key=True)
                                                           )
                 cc_ids.append([row.id, row.datatype])
+                import sys
+                print >>sys.stderr,row.datatype
                 if row.datatype == 'bool':
                     ccdict = {'__tablename__': 'custom_column_' + str(row.id),
                               'id': Column(Integer, primary_key=True),
                               'book': Column(Integer, ForeignKey('books.id')),
                               'value': Column(Boolean)}
+                elif row.datatype == 'int':
+                    ccdict = {'__tablename__': 'custom_column_' + str(row.id),
+                              'id': Column(Integer, primary_key=True),
+                              'book': Column(Integer, ForeignKey('books.id')),
+                              'value': Column(Integer)}
                 else:
                     ccdict = {'__tablename__': 'custom_column_' + str(row.id),
                               'id': Column(Integer, primary_key=True),
@@ -347,7 +354,7 @@ def setup_db():
                 cc_classes[row.id] = type('Custom_Column_' + str(row.id), (Base,), ccdict)
 
         for cc_id in cc_ids:
-            if cc_id[1] == 'bool':
+            if (cc_id[1] == 'bool') or (cc_id[1] == 'int'):
                 setattr(Books, 'custom_column_' + str(cc_id[0]), relationship(cc_classes[cc_id[0]],
                                                                            primaryjoin=(
                                                                            Books.id == cc_classes[cc_id[0]].book),

cps/templates/book_edit.html

@@ -68,6 +68,11 @@
                 <option value="False"  {% if book['custom_column_' ~ c.id]|length > 0 %}{% if book['custom_column_' ~ c.id][0].value ==false %}selected{% endif %}{% endif %}>{{_('No')}}</option>
               </select>
             {% endif %}
+
+            {% if c.datatype == 'int' %}
+              <input type="number" step="1" class="form-control" name="{{ 'custom_column_' ~ c.id }}" id="{{ 'custom_column_' ~ c.id }}" value="{{ book['custom_column_' ~ c.id][0].value }}">
+            {% endif %}
+
             {% if c.datatype in ['text', 'series'] and not c.is_multiple %}
               <input type="text" class="form-control" name="{{ 'custom_column_' ~ c.id }}" id="{{ 'custom_column_' ~ c.id }}"
               {% if book['custom_column_' ~ c.id]|length > 0 %}

cps/web.py

@@ -2589,6 +2589,22 @@ def edit_book(book_id):
                                     cc_class = db.cc_classes[c.id]
                                     new_cc = cc_class(value=to_save[cc_string], book=book_id)
                                     db.session.add(new_cc)
+                        elif c.datatype == 'int':
+                            if to_save[cc_string] == 'None':
+                                to_save[cc_string] = None
+                            if to_save[cc_string] != cc_db_value:
+                                if cc_db_value is not None:
+                                    if to_save[cc_string] is not None:
+                                        setattr(getattr(book, cc_string)[0], 'value', to_save[cc_string])
+                                    else:
+                                        del_cc = getattr(book, cc_string)[0]
+                                        getattr(book, cc_string).remove(del_cc)
+                                        db.session.delete(del_cc)
+                                else:
+                                    cc_class = db.cc_classes[c.id]
+                                    new_cc = cc_class(value=to_save[cc_string], book=book_id)
+                                    db.session.add(new_cc)
+
                         else:
                             if c.datatype == 'rating':
                                 to_save[cc_string] = str(int(float(to_save[cc_string]) * 2))