From c308cf7daaa4fa46377e2df0f2e9a397981e19b2 Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Fri, 10 Jun 2011 01:29:29 +0200
Subject: The current version is (mostly) working

---
 cli.py | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 76 insertions(+), 1 deletion(-)

(limited to 'cli.py')

diff --git a/cli.py b/cli.py
index d249917..fedb40c 100644
--- a/cli.py
+++ b/cli.py
@@ -3,7 +3,82 @@
     Metadata anonymisation toolkit
 """
 
-import lib
 import sys
+import mat
+import argparse
 
+__version__ = "0.1"
 
+def parsing():
+    '''
+	Parse the arguments,
+	and returns a dict
+    '''
+    parser = argparse.ArgumentParser(version=__version__,
+        description="Metadata Anonymisation Toolkit - CLI %s" % __version__)
+
+    #list and check clean are mutually exclusives
+    group = parser.add_mutually_exclusive_group()
+
+    #list meta
+    group.add_argument('--print-meta', '-p', action="store_true", default=False,
+        dest='just_list', help='List all the meta of a file,\
+        without removing them')
+
+    #check if the file is clean
+    group.add_argument('--check-clean', '-c', action="store_true",
+        default=False, dest='just_check',
+        help='Check if a file is clean of harmfull metadatas')
+
+    #list of files to process
+    parser.add_argument('filelist', action="store", type=str, nargs="+",
+        metavar='file', help='File(s) to process')
+
+    return parser.parse_args()
+
+def list_meta(class_file, filename):
+    '''
+	Print all the meta of "filename" on stdout
+    '''
+    print("[+] File %s :" % filename)
+    for key, item in class_file.get_meta().iteritems():
+        print("\t%s : %s" % (key, item) )
+
+def is_clean(class_file, filename):
+    '''
+	Say if "filename" is clean or not
+    '''
+    if class_file.is_clean():
+        print("[+] %s is clean" % filename)
+    else:
+        print("[+] %s is not clean" % filename)
+
+def clean_meta(class_file, filename):
+    '''
+	Clean the file "filename"
+    '''
+    print("[+] Cleaning %s" % filename)
+    if class_file.is_clean():
+        print("%s is already clean" % filename)
+    else:
+        class_file.remove_all()
+        print("%s cleaned !" % filename)
+
+def main():
+    args = parsing()
+
+    #func receive the function correponding to the options given as parameters
+    if args.just_list is True: #only print metadatas
+        func = list_meta
+    elif args.just_check is True: #only check if the file is clean
+        func = is_clean
+    else: #clean the file
+        func = clean_meta
+
+    for filename in args.filelist:
+        class_file = mat.create_class_file(filename)
+        func(class_file, filename)
+        print("\n")
+
+if __name__ == '__main__':
+    main()
-- 
cgit v1.3