Skip to content
Snippets Groups Projects
Unverified Commit 6fe4eb17 authored by James Curtin's avatar James Curtin Committed by GitHub
Browse files

Merge pull request #502 from cookiecutter-flask/fix-293

Support passing custom sets of kwargs when creating a reference column
parents 0146d26d 90ed8fec
No related branches found
No related tags found
No related merge requests found
......@@ -62,7 +62,7 @@ class SurrogatePK(object):
return None
def reference_col(tablename, nullable=False, pk_name='id', **kwargs):
def reference_col(tablename, nullable=False, pk_name='id', foreign_key_kwargs=None, column_kwargs=None):
"""Column that adds primary key foreign key reference.
Usage: ::
......@@ -70,6 +70,9 @@ def reference_col(tablename, nullable=False, pk_name='id', **kwargs):
category_id = reference_col('category')
category = relationship('Category', backref='categories')
"""
foreign_key_kwargs = foreign_key_kwargs or {}
column_kwargs = column_kwargs or {}
return Column(
db.ForeignKey('{0}.{1}'.format(tablename, pk_name)),
nullable=nullable, **kwargs)
db.ForeignKey('{0}.{1}'.format(tablename, pk_name), **foreign_key_kwargs),
nullable=nullable, **column_kwargs)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment