From d8a6ae28ff739860fe391d754c384ab2aa2bb821 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Tue, 11 Aug 2015 17:14:16 +0200
Subject: [PATCH] Add test scripts to get and update an event.

---
 examples/get.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 examples/up.py  | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100755 examples/get.py
 create mode 100755 examples/up.py

diff --git a/examples/get.py b/examples/get.py
new file mode 100755
index 0000000..17f4429
--- /dev/null
+++ b/examples/get.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from pymisp import PyMISP
+from keys import url_priv, key_priv
+# from keys import url_cert, key_cert
+import argparse
+import os
+import json
+
+
+# Usage for pipe masters: ./last.py -l 5h | jq .
+
+
+def init(url, key):
+    return PyMISP(url, key, True, 'json')
+
+
+def get_event(m, event, out=None):
+    result = m.get_event(event)
+    r = result.json()
+    if out is None:
+        print(json.dumps(r['Event']) + '\n')
+    else:
+        with open(out, 'w') as f:
+            f.write(json.dumps(r['Event']) + '\n')
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Get an event from a MISP instance.')
+    parser.add_argument("-e", "--event", required=True, help="Event ID to get.")
+    parser.add_argument("-o", "--output", help="Output file")
+
+    args = parser.parse_args()
+
+    if args.output is not None and os.path.exists(args.output):
+        print('Output file already exists, abord.')
+        exit(0)
+
+    misp = init(url_priv, key_priv)
+    # misp = init(url_cert, key_cert)
+
+    get_event(misp, args.event, args.output)
diff --git a/examples/up.py b/examples/up.py
new file mode 100755
index 0000000..d54aacb
--- /dev/null
+++ b/examples/up.py
@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from pymisp import PyMISP
+from keys import url_priv, key_priv
+# from keys import url_cert, key_cert
+import argparse
+
+
+# Usage for pipe masters: ./last.py -l 5h | jq .
+
+
+def init(url, key):
+    return PyMISP(url, key, True, 'json')
+
+
+def up_event(m, event, content):
+    with open(content, 'r') as f:
+        result = m.update_event(event, f.read())
+    print result.text
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Get an event from a MISP instance.')
+    parser.add_argument("-e", "--event", required=True, help="Event ID to get.")
+    parser.add_argument("-i", "--input", required=True, help="Input file")
+
+    args = parser.parse_args()
+
+    misp = init(url_priv, key_priv)
+    # misp = init(url_cert, key_cert)
+
+    up_event(misp, args.event, args.input)