def loadVideosCategories(catalog): videocategoriesfile = cf.data_dir + 'videos-small.csv' input_file = csv.DictReader(open(videocategoriesfile, encoding='utf-8')) for videocategory in input_file: model.addVideoCategory(catalog, videocategory)
def loadVideos(catalog): videosfile = cf.data_dir + 'Videos/videos-large.csv' input_file = csv.DictReader(open(videosfile, encoding='utf-8')) contador = 1 for e in input_file: ee = { 'video_id': e['video_id'], 'trending_date': datetime.strptime(e['trending_date'], '%y.%d.%m').date(), 'title': e['title'], 'channel_title': e['title'], 'category_id': e['category_id'], 'publish_time': iso.parse_date(e['publish_time']), 'tags': e['tags'], 'views': e['views'], 'likes': e['likes'], 'dislikes': e['dislikes'], 'country': e['country'] } model.addVideo(catalog, ee) model.addVideoCountry(catalog, ee, contador) model.addVideoCategory(catalog, ee, contador) contador += 1
def loadVideos(catalog): """ Args: catalog: Catálogo de videos. Carga todos los vídeos del archivo y los agrega a la lista de vídeos """ videosfile = cf.data_dir + 'videos/videos-large.csv' # videos-large para la entrega input_file = csv.DictReader(open(videosfile, encoding='utf-8')) video1 = None size = 0 for video in input_file: filtered_video = { 'video_id': video['video_id'], 'trending_date': dt.datetime.strptime(video['trending_date'], '%y.%d.%m').date(), 'title': video['title'], 'channel_title': video['channel_title'], 'category_id': int(video['category_id']), 'publish_time': dt.datetime.strptime(video['publish_time'], "%Y-%m-%dT%H:%M:%S.%fZ"), 'tags': video['tags'].replace("\"", ''), 'views': int(video['views']), 'likes': int(video['likes']), 'dislikes': int(video['dislikes']), 'country': video['country'], 'dias_t': 1 } if size == 0: video1 = filtered_video size += 1 model.addVideoCountry(catalog, filtered_video['country'], filtered_video) model.addVideoCategory(catalog, filtered_video['category_id'], filtered_video) return video1, size
def loadVideoCategories(catalog): idsfile = cf.data_dir + 'category-id.csv' input_file = csv.DictReader(open(idsfile, encoding='utf-8')) for id in input_file: model.addVideoCategory(catalog, id)
def loadVideos(catalog): videosfile = cf.data_dir + 'Samples/videos-large.csv' input_file = csv.DictReader(open(videosfile, encoding='utf-8')) for video in input_file: model.addVideoCategory(catalog, video)