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')