def get_date()

in src/scripts/extract_data.py [0:0]

• 17 lines of code
• 8 McCabe index (conditional complexity)

def get_date(dataset_name, keys):
    date = {}
    if dataset_name == 'gcn':
        with open('./data/gcn/dataset.json') as f:
            data = json.load(f)
        for key, body in data.items():
            if key in keys:
                body = body.split('\n')
                date_str = list(filter(lambda x: x.startswith("DATE:"), body))
                date[key] = date_str[0].replace("DATE:", '').strip() if len(date_str) > 0 else ""
    elif dataset_name == 'atel':
        with open('./data/atel/dataset.csv', 'r') as f:
            reader = csv.DictReader(f)
            for row in reader:
                if row['id'] in keys:
                    date[row['id']] = row['date']
    return date