aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-17 14:04:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add names_only parameter to get_columns_for_table (#2219)

Browse Source
This commit is contained in:
Kevin Chung
2022-11-05 18:12:19 -04:00
committed by GitHub
parent 5daa85fce6
commit 95bfb96a82
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
CTFd/plugins/migrations.py
View File

@@ -22,12 +22,14 @@ def get_all_tables(op):
return tables
def get_columns_for_table(op, table_name):
def get_columns_for_table(op, table_name, names_only=False):
"""
Function to list the columns in a table from a migration
"""
inspector = SQLAInspect(op.get_bind())
columns = inspector.get_columns(table_name)
if names_only is True:
columns = [c["name"] for c in columns]
return columns