Spaces:
Sleeping
Sleeping
File size: 2,963 Bytes
a0e7778 e733251 a0e7778 f4fb0fb a0e7778 f4fb0fb a0e7778 f4fb0fb a0e7778 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import yaml
import json
from pathlib import Path
import pandas as pd
from src.compliance_analysis import check_overall_compliance
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
files = ["./project_cc.yaml", "./data_cc.yaml", "./data_cc.yaml", "./model_cc.yaml", "./model_cc.yaml", "./model_cc.yaml"]
# def load_data(files):
# cards = []
# for file in files:
# with open(file, 'r') as f:
# if Path(f.name).name == "project_cc.yaml":
# content = f.read()
# project_cc_yaml = yaml.safe_load(content)
# data = project_cc_yaml
# card_type = "project"
# cards.append((card_type, data))
# if Path(f.name).name == "data_cc.yaml":
# data_cc_yaml = yaml.safe_load(content)
# data = data_cc_yaml
# card_type = "data"
# cards.append((card_type, data))
# if Path(f.name).name == "model_cc.yaml":
# model_cc_yaml = yaml.safe_load(content)
# data = model_cc_yaml
# card_type = "model"
# cards.append((card_type, data))
# return cards
# cards = load_data(files)
def gather_cards(files):
cards = {}
cards['project_file'] = ''
cards['data_files'] = []
cards['model_files'] = []
for file in files:
with open(file, 'r') as f:
content = yaml.safe_load(f.read())
if content['card_type'] == "project":
cards['project_file'] = f.name
if content['card_type'] == "data":
cards['data_files'].append(f.name)
if content['card_type'] == "model":
cards['model_files'].append(f.name)
return cards
cards = gather_cards(files)
# def load_data(files):
# cards = []
# for file in files:
# with open(file, 'r') as f:
# if Path(f.name).name == "project_cc.yaml":
# content = f.read()
# pcrojet_cc_yaml = yaml.safe_load(content)
# data = project_cc_yaml
# card_type = "project"
# cards.append((card_type, data))
# if Path(f.name).name == "data_cc.yaml":
# data_cc_yaml = yaml.safe_load(content)
# data = data_cc_yaml
# card_type = "data"
# cards.append((card_type, data))
# if Path(f.name).name == "model_cc.yaml":
# model_cc_yaml = yaml.safe_load(content)
# data = model_cc_yaml
# card_type = "model"
# cards.append((card_type, data))
# return cards
def compliance_analysis(cards):
results = []
dispositive_variables = check_overall_compliance(cards)
results.append(dispositive_variables)#['msg'])
return results
print(json.dumps(compliance_analysis(cards), indent=4,))
|