Commit 9587bf0 1 parent 94f43b6 commit 9587bf0 Copy full SHA for 9587bf0
File tree 1 file changed +16
-9
lines changed
1 file changed +16
-9
lines changed Original file line number Diff line number Diff line change 12
12
from fides .api .db .session import get_db_engine , get_db_session
13
13
from fides .api .models .sql_models import DataCategory as DataCategoryDbModel
14
14
from fides .api .tasks .scheduled .scheduler import async_scheduler , scheduler
15
+ from fides .api .util .data_category import get_data_categories_from_db
15
16
from tests .conftest import create_citext_extension
16
17
17
18
@@ -62,17 +63,23 @@ def fideslang_data_categories(db):
62
63
"""
63
64
Creates a database record for each data category in the fideslang taxonomy.
64
65
"""
65
- cats = []
66
- for obj in DEFAULT_TAXONOMY .data_category :
67
- try :
68
- cats .append (DataCategoryDbModel .from_fideslang_obj (obj ).save (db ))
69
- except IntegrityError :
70
- pass
71
66
72
- yield cats
67
+ categories = []
68
+ existing_categories = get_data_categories_from_db (db )
69
+
70
+ for default_category in DEFAULT_TAXONOMY .data_category :
71
+ if default_category .fides_key not in existing_categories :
72
+ try :
73
+ categories .append (
74
+ DataCategoryDbModel .from_fideslang_obj (default_category ).save (db )
75
+ )
76
+ except IntegrityError :
77
+ pass
78
+
79
+ yield categories
73
80
74
- for cat in cats :
81
+ for category in categories :
75
82
try :
76
- cat .delete (db )
83
+ category .delete (db )
77
84
except ObjectDeletedError :
78
85
pass
You can’t perform that action at this time.
0 commit comments