Added arguments parser
This commit is contained in:
15
main.py
15
main.py
@@ -1,3 +1,4 @@
|
|||||||
|
import argparse
|
||||||
import asyncio
|
import asyncio
|
||||||
import itertools
|
import itertools
|
||||||
import logging
|
import logging
|
||||||
@@ -71,10 +72,15 @@ class TrackInfo:
|
|||||||
Track Number: {self.__number}"""
|
Track Number: {self.__number}"""
|
||||||
|
|
||||||
|
|
||||||
async def main() -> Any:
|
def arg_parser():
|
||||||
|
parser = argparse.ArgumentParser(description="Organize music from source folder to destination folder using Shazam")
|
||||||
|
parser.add_argument("--src", "-s", metavar="SRC", type=str, help="Source directory to scan files")
|
||||||
|
parser.add_argument("--dest", "-d", metavar="DEST", type=str, help="Destination directory to copy files")
|
||||||
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
|
async def main(src, dst) -> Any:
|
||||||
shazam: Shazam = Shazam()
|
shazam: Shazam = Shazam()
|
||||||
src: str = input("Which path scan?: ")
|
|
||||||
dst: str = input("Where do you want to copy files?: ")
|
|
||||||
for song_file in ioutil.scan_folder(src):
|
for song_file in ioutil.scan_folder(src):
|
||||||
if ioutil.accepted_file_type(song_file):
|
if ioutil.accepted_file_type(song_file):
|
||||||
shazam_metadata: dict[str, Any] = await shazam.recognize(song_file)
|
shazam_metadata: dict[str, Any] = await shazam.recognize(song_file)
|
||||||
@@ -92,5 +98,6 @@ async def main() -> Any:
|
|||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
logging.basicConfig(filename="music.log", level=logging.DEBUG)
|
logging.basicConfig(filename="music.log", level=logging.DEBUG)
|
||||||
|
args = arg_parser()
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
loop.run_until_complete(main())
|
loop.run_until_complete(main(args.src, args.dest))
|
||||||
|
|||||||
Reference in New Issue
Block a user