def test_courses_unique_across_terms(): cal = AcademicCalendar('ualberta') db.drop_all() db.create_all() terms = cal._fetch(datatype='terms') cal._save(terms, datatype='terms') courses = cal._fetch(datatype='courses', term='1510') cal._save(courses, datatype='courses') courses_overlapping = cal._fetch(datatype='courses', term='1520') cal._save(courses_overlapping, datatype='courses') assert len(Course.query.all()) == len(courses) + len(courses_overlapping)
def fill_institutions(search_params=None): #pylint: disable=W0613 db.create_all() if Institution.query.first() is None: config_file = os.path.join(classtime.brain.institutions.CONFIG_FOLDER_PATH, 'institutions.json') with open(config_file, 'r') as config: config = json.loads(config.read()) institutions = config.get('institutions') for institution in institutions: if not Institution.query.get(institution.get('institution')): db.session.add(Institution(institution)) try: db.session.commit() except: logging.error('Institutions failed to add to database') return None
def fill_institutions(search_params=None): #pylint: disable=W0613 db.create_all() if Institution.query.first() is None: config_file = os.path.join( classtime.brain.institutions.CONFIG_FOLDER_PATH, 'institutions.json') with open(config_file, 'r') as config: config = json.loads(config.read()) institutions = config.get('institutions') for institution in institutions: if not Institution.query.get(institution.get('institution')): db.session.add(Institution(institution)) try: db.session.commit() except: logging.error('Institutions failed to add to database') return None
def create(self): """Create the database, if it did not already exist """ db.create_all()
def create_db(): db.create_all() logging.info('DB created!')
def setup_class(cls): app.config['TESTING'] = True cls.client = app.test_client() db.create_all()