oshipka/vm_gen/templates/_model_py

[%- if _choice_types %]
from sqlalchemy_utils import ChoiceType
[%- endif %]

class [[ name ]](db.Model, ModelController[% for inherit in interits %], [[ inherit ]][% endfor %]):
    [%- include "_model_choice_header_py" %]
    [%- include "_model_searchable_header_py" %]

    _file_columns = [ [%- for column in columns %][%- if column.is_file %]"[[ column.name ]]"[%- endif %] [%- endfor %] ]

    [%- for column in columns %]
      [%- if column._type == 'relationship' %]
    [%- include "_relationship_py" %]
      [%- else %]
    [[ column.name ]] = db.Column([[ column._type ]],
                                  [%- if column.default %]default="[[ column.default ]]",[%- endif %]
                                  [%- if column.index %]index=True,[%- endif %])
      [%- endif %]
    [%- endfor %]

    [%- if extra_code %]

    [[ extra_code ]]
    [%- endif %]

    [%- if display %]
    def __repr__(self):
        [%- if display.tertiary %]
        return "{} ({} - {})".format(self.[[ display.primary ]], self.[[ display.secondary ]], self.[[ display.tertiary]])
        [%- elif display.secondary %]
        return "{} ({})".format(self.[[ display.primary ]], self.[[ display.secondary ]])
        [%- else %]
        return "{}".format(self.[[ display.primary ]])
        [%- endif %]
    [%- endif %]