Initial commit of merge_packages script

ikiril01 · ikiril01 · commit dd539b1214f5 · 2014-07-07T13:16:56.000-04:00
diff --git a/scripts/merge_packages.py b/scripts/merge_packages.py
@@ -0,0 +1,59 @@
+# merge_packages script
+# v0.10 BETA
+# Merges two or more MAEC Package documents (.xml files)
+# Attempts to merge related Malware Subjects
+import sys
+import os
+import maec
+from maec.utils.merge import merge_documents
+
+USAGE_TEXT = """
+MAEC Package Merge Script v0.10 BETA
+   *Merges two or more MAEC Package XML documents
+   *Attempts to merge related (e.g., same MD5 hash) Malware Subjects
+
+Usage: python merge_packages.py -o <output file name> -l <single whitespace separated list of MAEC Package files> OR -d <directory name>
+"""
+
+def main():
+    infilenames = []
+    list_mode = False
+    directoryname = ''
+    outfilename = ''
+
+    #Get the command-line arguments
+    args = sys.argv[1:]
+    
+    if len(args) < 3:
+        print USAGE_TEXT
+        sys.exit(1)
+        
+    for i in range(0,len(args)):
+        if args[i] == '-o':
+            outfilename = args[i+1]
+        elif args[i] == '-l':
+            list_mode = True
+        elif args[i] == '-d':
+            directoryname = args[i+1]
+
+    if outfilename == '':
+        print USAGE_TEXT
+        sys.exit(1)
+
+    sys.stdout.write("Merging...")
+    # Get the list of input files and perform the merge operation
+    if list_mode:
+        files = args[3:]
+        merge_documents(files, outfilename)
+    elif directoryname != '':
+        file_list = []
+        for filename in os.listdir(directoryname):
+            if '.xml' not in filename:
+                pass
+            else:
+                file_list.append(os.path.join(directoryname, filename))
+        merge_documents(file_list, outfilename)
+    sys.stdout.write("Done.")
+
+if __name__ == "__main__":
+    main()    
