def test_upload_video(self): """Test uploading a video file using mocks""" youtube = YoutubeService() youtube.upload_video = Mock(return_value=(Response.SUCCESS, None)) response_code, response = youtube.upload_video(self.SAMPLE_VIDEO) youtube.upload_video.assert_called_with(self.SAMPLE_VIDEO) self.assertEqual(Response.SUCCESS, response_code)
def upload(files, token, assume_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 assume_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)) else: # Gather videos specified and vids from folders specified into list videos = gather_videos(files) # Now begin upload process if not videos: print("Nothing to upload") # Prompt for confirmation elif prompt_user(videos, confirmation=assume_yes): youtube_service = YoutubeService() # Authorize with OAuth2 token youtube_service.authorize(token) for video in videos: response_code, response = youtube_service.upload_video(video) handle_response(response_code, response) # Response was no, so do nothing else: print("Exiting...")
def upload(files, token, assume_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 assume_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)) else: # Gather videos specified and vids from folders specified into list videos = gather_videos(files) # Now begin upload process if not videos: print("Nothing to upload") # Prompt for confirmation elif prompt_user(videos, confirmation=assume_yes): youtube_service = YoutubeService() # Authorize with OAuth2 token youtube_service.authorize(token) for video in videos: response_code, response = youtube_service.upload_video(video) handle_response(response_code, response) # Response was no, so do nothing else: print("Exiting...")
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 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 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))