def upload(self, files, token, yes): """Uploads a file(s) to YouTube using the YouTube service API This function uploads a list of videos and/or directories of videos to YouTube. Args: token - location of an oauth2 token files - list of files and directories to upload yes - if True, assume yes to all interaction (default: False) """ # check if token exists if not os.path.exists(token): print( "{} does not exist, please specify a valid token file".format( token)) return # gather videos specified and vids from folders specified into list videos = [] for item in files: # crawl subfolders if os.path.isdir(item): for root, dirs, files_ in os.walk(item): for f in files_: filepath = os.path.join(root, f) # check if its a video and make sure its not a duplicate if YoutubeService.valid_video_file( filepath) and filepath not in videos: videos.append(filepath) # if it exists it is a single file, check if its a video, and make sure its not a duplicate elif os.path.exists(item) and YoutubeService.valid_video_file( item) and item not in videos: videos.append(item) # now begin upload process if not videos: print("Nothing to upload") else: youtube_service = YoutubeService() # authorize with OAuth2 token youtube_service.authorize(token) if not yes: print("Found videos:") print("\n".join(videos)) question = "Are you sure you would like to upload these videos? [Y/n] " if yes or raw_input(question).lower() in ('', 'y', 'yes'): for video in videos: self.handle_response(youtube_service.upload_video(video))
def gather_videos(files): """Gather all valid videos into a set for upload""" # Because we are using a set, no duplicates will be present videos = set() for item in files: # Crawl subfolders if os.path.isdir(item): for root, _, filenames in os.walk(item): for filename in filenames: filepath = os.path.join(root, filename) # Check if its a video if YoutubeService.valid_video_file(filepath): videos.add(filepath) # If it exists it is a single file, check if its a video elif os.path.exists(item) and YoutubeService.valid_video_file(item): videos.add(item) return videos
def gather_videos(files): """Gather all valid videos into a set for upload""" # Because we are using a set, no duplicates will be present videos = set() for item in files: # Crawl subfolders if os.path.isdir(item): for root, _, filenames in os.walk(item): for filename in filenames: filepath = os.path.join(root, filename) # Check if its a video if YoutubeService.valid_video_file(filepath): videos.add(filepath) # If it exists it is a single file, check if its a video elif os.path.exists(item) and YoutubeService.valid_video_file(item): videos.add(item) return videos
def upload(self, files, token, yes): """Uploads a file(s) to YouTube using the YouTube service API This function uploads a list of videos and/or directories of videos to YouTube. Args: token - location of an oauth2 token files - list of files and directories to upload yes - if True, assume yes to all interaction (default: False) """ # check if token exists if not os.path.exists(token): print("{} does not exist, please specify a valid token file".format(token)) return # gather videos specified and vids from folders specified into list videos = [] for item in files: # crawl subfolders if os.path.isdir(item): for root, dirs, files_ in os.walk(item): for f in files_: filepath = os.path.join(root, f) # check if its a video and make sure its not a duplicate if YoutubeService.valid_video_file(filepath) and filepath not in videos: videos.append(filepath) # if it exists it is a single file, check if its a video, and make sure its not a duplicate elif os.path.exists(item) and YoutubeService.valid_video_file(item) and item not in videos: videos.append(item) # now begin upload process if not videos: print("Nothing to upload") else: youtube_service = YoutubeService() # authorize with OAuth2 token youtube_service.authorize(token) if not yes: print("Found videos:") print("\n".join(videos)) question = "Are you sure you would like to upload these videos? [Y/n] " if yes or raw_input(question).lower() in ('', 'y', 'yes'): for video in videos: self.handle_response(youtube_service.upload_video(video))
def cmd_line(self, argv): """Initializes command line interface Args: argv: command line arguments to parse """ self.parser.add_argument("-c", "--client_secrets", help="Path to client secrets file", default=self.client_secrets) self.parser.add_argument("-t", "--token", help="Path to OAuth2 token", default=self.oauth2_token) self.parser.add_argument("-f", "--files", help="Path to file or filefolder for upload", default=self.video_directory) self.parser.add_argument("-y", "--yes", help="Assume yes", action="store_true") flags = self.parser.parse_args(argv[1:]) youtube_service = YoutubeService() # command line arguments neede for Google Python Client need to always be passed in for authentication youtube_service.authenticate(flags.client_secrets, flags.token, flags) # path doesn't exist, print an error message if not os.path.exists(flags.files): print "{} does not exit".format(flags.files) # path is a folder, upload all videos in the folder elif os.path.isdir(flags.files): self.upload_folder_cmd_line(youtube_service, flags.files, flags.yes) # single file check if it is an uploadable video file elif youtube_service.valid_video_file(flags.files): if flags.yes: self.handle_response(youtube_service.upload_video(flags.files)) else: response = raw_input( "Are you sure you would like to upload the following file? [Y/n]\n" + flags.files + "\n") if response.lower() in ('', 'y', 'yes'): self.handle_response(youtube_service.upload_video(flags.files)) # if this case is reached it means there is nothing to upload else: print "Nothing to upload"
def test_valid_video_file(video, expected): """Tests valid_video_file function for all test cases.""" assert YoutubeService.valid_video_file(video) == expected