def populate_raw_data_types(drop): """ Populate default set of raw data types into the system. :param drop: Drop the existing collection before trying to populate. :type: boolean """ # define your raw data types here data_types = ['Text', 'JSON'] if drop: RawDataType.drop_collection() if len(RawDataType.objects()) < 1: for data_type in data_types: dt = RawDataType() dt.name = data_type dt.save() print "Raw Data Types: added %s types!" % len(data_types) else: print "Raw Data Types: existing documents detected. skipping!"