social-clusters/prepare_dataset/school.py

19 lines
729 B
Python

from prepare_dataset.raw_data import RawData
class School:
def __init__(self, raw_school):
data = RawData(raw_school)
self.id = data.get_int('id')
self.name = data.get_str('name')
self.country_id = RawData.get_int_st(data.country, 'id')
self.city_id = RawData.get_int_st(data.city, 'id')
self.year_from = data.get_int('year_from')
self.year_to = data.get_int('year_to')
self.year_graduated = data.get_int('year_graduated')
self.speciality = data.get_str('speciality')
self.type_id = data.get_int('type')
self.type = data.get_str('type_str')
self.litera_id = data.get_int('class_id')
self.litera = data.get_str('class')