generated from simonw/python-lib-template-repository
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add __init__.py to tests folder and move main logic into main() (#12)
* added empty __init__.py file to tests folder, moved main logic into main() * update to v0.1.1
- Loading branch information
Showing
3 changed files
with
107 additions
and
103 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,108 +1,112 @@ | ||
| import argparse | ||
| import os | ||
|
|
||
| images = [".jpeg", ".jpg", ".png", ".gif"] | ||
| videos = [".mp4", ".flv", ".m4v", ".webm"] | ||
| text = [ ".txt", ".md", ".doc", ".docx", ".pdf"] | ||
| imageCount = 0 | ||
| movedImagesCount = 0 | ||
| videoCount = 0 | ||
| movedVideosCount = 0 | ||
| textCount = 0 | ||
| movedTextFilesCount = 0 | ||
| def main(): | ||
| images = [".jpeg", ".jpg", ".png", ".gif"] | ||
| videos = [".mp4", ".flv", ".m4v", ".webm"] | ||
| text = [ ".txt", ".md", ".doc", ".docx", ".pdf"] | ||
| imageCount = 0 | ||
| movedImagesCount = 0 | ||
| videoCount = 0 | ||
| movedVideosCount = 0 | ||
| textCount = 0 | ||
| movedTextFilesCount = 0 | ||
|
|
||
| parser = argparse.ArgumentParser( | ||
| prog="folderizer", | ||
| description="File organization command-line tool.", | ||
| epilog="Thanks for using %(prog)s!") | ||
| parser.add_argument('filepath') | ||
| parser.add_argument("-v","--version", action="version", version="%(prog)s 0.1.0") | ||
| parser.add_argument("-d", "--default", action='store_true', help="Sorts files into folders by category (image, video, text, code, etc)") | ||
| args = parser.parse_args() | ||
| print(args, args.filepath, args.default) | ||
| if args.default: | ||
| print("Default behaviour") | ||
| print(os.path.abspath(args.filepath)) | ||
| target_path = os.path.abspath(args.filepath) | ||
| print("Target Path: " + target_path) | ||
| if (os.access(target_path, os.F_OK)): | ||
| print("Filepath exists") | ||
| print(os.scandir(target_path)) | ||
| with os.scandir(target_path) as it: | ||
| for entry in it: | ||
| if not entry.name.startswith('.') and entry.is_file(): | ||
| print("Entry name: " + entry.name) | ||
| print(os.path.splitext(entry.name)[1]) | ||
| if os.path.splitext(entry.name)[1] in images: | ||
| imageCount+=1 | ||
| print("image detected") | ||
| imagePath = os.path.join(target_path, "images") | ||
| print(imagePath) | ||
| if not (os.access(imagePath, os.F_OK)): | ||
| print('creating images folder') | ||
| os.mkdir(imagePath) | ||
| else: | ||
| print("not creating images folder") | ||
| # move file to images folder | ||
| print("Image file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = imagePath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedImagesCount+=1 | ||
| elif os.path.splitext(entry.name)[1] in videos: | ||
| videoCount+=1 | ||
| print("video detected") | ||
| videoPath = os.path.join(target_path, "videos") | ||
| print(videoPath) | ||
| if not (os.access(videoPath, os.F_OK)): | ||
| print('creating videos folder!') | ||
| os.mkdir(videoPath) | ||
| else: | ||
| print("not creating videos folder") | ||
| # move file to videos folder | ||
| print("Video file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = videoPath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedVideosCount+=1 | ||
| elif os.path.splitext(entry.name)[1] in text: | ||
| textCount+=1 | ||
| print("text file detected") | ||
| textPath = os.path.join(target_path, "textFiles") | ||
| print(textPath) | ||
| if not (os.access(textPath, os.F_OK)): | ||
| print('creating textFiles folder!') | ||
| os.mkdir(textPath) | ||
| else: | ||
| print("not creating textFiles folder") | ||
| # move file to videos folder | ||
| print("Text file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = textPath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedTextFilesCount+=1 | ||
| print('------------------------------------') | ||
| print('SUMMARY') | ||
| print('------------------------------------') | ||
| print('Images found: ',imageCount) | ||
| print('Images successfully moved: ', movedImagesCount) | ||
| print('Videos found: ',videoCount) | ||
| print('Videos successfully moved: ', movedVideosCount) | ||
| print('Text files found: ',textCount) | ||
| print('Text files successfully moved: ', movedTextFilesCount) | ||
| else: | ||
| print("Filepath doesn't exist") | ||
| parser = argparse.ArgumentParser( | ||
| prog="folderizer", | ||
| description="File organization command-line tool.", | ||
| epilog="Thanks for using %(prog)s!") | ||
| parser.add_argument('filepath') | ||
| parser.add_argument("-v","--version", action="version", version="%(prog)s 0.1.1") | ||
| parser.add_argument("-d", "--default", action='store_true', help="Sorts files into folders by category (image, video, text, code, etc)") | ||
| args = parser.parse_args() | ||
| print(args, args.filepath, args.default) | ||
| if args.default: | ||
| print("Default behaviour") | ||
| print(os.path.abspath(args.filepath)) | ||
| target_path = os.path.abspath(args.filepath) | ||
| print("Target Path: " + target_path) | ||
| if (os.access(target_path, os.F_OK)): | ||
| print("Filepath exists") | ||
| print(os.scandir(target_path)) | ||
| with os.scandir(target_path) as it: | ||
| for entry in it: | ||
| if not entry.name.startswith('.') and entry.is_file(): | ||
| print("Entry name: " + entry.name) | ||
| print(os.path.splitext(entry.name)[1]) | ||
| if os.path.splitext(entry.name)[1] in images: | ||
| imageCount+=1 | ||
| print("image detected") | ||
| imagePath = os.path.join(target_path, "images") | ||
| print(imagePath) | ||
| if not (os.access(imagePath, os.F_OK)): | ||
| print('creating images folder') | ||
| os.mkdir(imagePath) | ||
| else: | ||
| print("not creating images folder") | ||
| # move file to images folder | ||
| print("Image file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = imagePath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedImagesCount+=1 | ||
| elif os.path.splitext(entry.name)[1] in videos: | ||
| videoCount+=1 | ||
| print("video detected") | ||
| videoPath = os.path.join(target_path, "videos") | ||
| print(videoPath) | ||
| if not (os.access(videoPath, os.F_OK)): | ||
| print('creating videos folder!') | ||
| os.mkdir(videoPath) | ||
| else: | ||
| print("not creating videos folder") | ||
| # move file to videos folder | ||
| print("Video file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = videoPath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedVideosCount+=1 | ||
| elif os.path.splitext(entry.name)[1] in text: | ||
| textCount+=1 | ||
| print("text file detected") | ||
| textPath = os.path.join(target_path, "textFiles") | ||
| print(textPath) | ||
| if not (os.access(textPath, os.F_OK)): | ||
| print('creating textFiles folder!') | ||
| os.mkdir(textPath) | ||
| else: | ||
| print("not creating textFiles folder") | ||
| # move file to videos folder | ||
| print("Text file name: " + entry.name) | ||
| print(entry) | ||
| oldPath = target_path + "\\" + entry.name | ||
| newPath = textPath + "\\" + entry.name | ||
| print("old filepath: " + oldPath) | ||
| print("new filepath: " + newPath) | ||
| # attempt rename | ||
| os.rename(oldPath, newPath) | ||
| movedTextFilesCount+=1 | ||
| print('------------------------------------') | ||
| print('SUMMARY') | ||
| print('------------------------------------') | ||
| print('Images found: ',imageCount) | ||
| print('Images successfully moved: ', movedImagesCount) | ||
| print('Videos found: ',videoCount) | ||
| print('Videos successfully moved: ', movedVideosCount) | ||
| print('Text files found: ',textCount) | ||
| print('Text files successfully moved: ', movedTextFilesCount) | ||
| else: | ||
| print("Filepath doesn't exist") | ||
| def example_function(): | ||
| return 1 + 1 | ||
|
|
||
| if __name__ == "__main__": | ||
| main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.