diff --git a/fs/opt/groomer/functions.py b/fs/opt/groomer/functions.py index 496dd2a..e015923 100644 --- a/fs/opt/groomer/functions.py +++ b/fs/opt/groomer/functions.py @@ -7,7 +7,7 @@ import shlex import subprocess import time -from helpers import FileBase, KittenGroomerBase +from helpers import FileBase, KittenGroomerBase, main LIBREOFFICE = '/usr/bin/unoconv' GS = '/usr/bin/gs' @@ -318,6 +318,4 @@ class KittenGroomer(KittenGroomerBase): self._print_log() if __name__ == '__main__': - kg = KittenGroomer('/home/raphael/gits/KittenGroomer/tests/content_img_vfat_norm', - '/home/raphael/gits/KittenGroomer/tests/content_img_vfat_norm_out') - kg.processdir() + main(KittenGroomer) diff --git a/fs/opt/groomer/functions_pier9.py b/fs/opt/groomer/functions_pier9.py index 4476c95..a3eb577 100644 --- a/fs/opt/groomer/functions_pier9.py +++ b/fs/opt/groomer/functions_pier9.py @@ -2,7 +2,7 @@ # -*- coding: utf-8 -*- import os -from helpers import FileBase, KittenGroomerBase +from helpers import FileBase, KittenGroomerBase, main printers = ['.STL', '.obj'] @@ -65,6 +65,4 @@ class KittenGroomerPier9(KittenGroomerBase): if __name__ == '__main__': - kg = KittenGroomerPier9('/home/raphael/gits/KittenGroomer/tests/content_img_vfat_norm', - '/home/raphael/gits/KittenGroomer/tests/content_img_vfat_norm_out') - kg.processdir() + main(KittenGroomerPier9) diff --git a/fs/opt/groomer/helpers.py b/fs/opt/groomer/helpers.py index fd72d5a..e1d5fe8 100644 --- a/fs/opt/groomer/helpers.py +++ b/fs/opt/groomer/helpers.py @@ -3,7 +3,7 @@ import os import shutil from twiggy import quickSetup, log - +import argparse class KittenGroomerError(Exception): def __init__(self, message): @@ -138,3 +138,12 @@ class KittenGroomerBase(object): Main function doing the work, you have to implement it yourself. ''' raise ImplementationRequired('You have to implement the result processdir.') + + +def main(kg_implementation): + parser = argparse.ArgumentParser(prog='KittenGroomer', description='Call the KittenGroomer implementation to do things on files present in the source directory to the destination directory') + parser.add_argument('-s', '--source', type=str, help='Source directory') + parser.add_argument('-d', '--destination', type=str, help='Destination directory') + args = parser.parse_args() + kg = kg_implementation(args.source, args.destination) + kg.processdir()