Implement simple storage in the filesystem

2 files changed, 28 insertions, 2 deletions

diff --git a/flask_app.py b/flask_app.py
index a69f4f8..87697e5 100644
--- a/flask_app.py
+++ b/flask_app.py
@@ -3,10 +3,12 @@ The web application built on Flask is contained within this file.
 When run as a script, the Flask development server is started.
 """
 
-import submission_pb2
+import os
+import submission_pb2, storage
 from flask import Flask, request
 
 app = Flask(__name__)
+store = storage.FilesystemStorage('logs/')
 
 @app.route('/')
 def index():
@@ -20,7 +22,8 @@ def submit():
 	"""
 	submission = submission_pb2.Submission()
 	submission.ParseFromString(request.data)
-	return str(submission)
+	store.save_file(request.remote_addr, submission.filename, submission.data)
+	return ''
 
 if __name__ == '__main__':
 	app.run(host='::1', debug=True)
diff --git a/storage.py b/storage.py
new file mode 100644
index 0000000..847806f
--- /dev/null
+++ b/storage.py
@@ -0,0 +1,23 @@
+"""
+Implements storage of collected log files in the local filesystem.
+"""
+
+import os, errno
+
+class FilesystemStorage:
+	def __init__(self, root):
+		self.root = root
+		try:
+			os.mkdir(root)
+		except OSError:
+			pass # TODO: proper handling
+
+	def save_file(self, source, filename, data):
+		try:
+			os.mkdir(os.path.join(self.root, source))
+		except OSError:
+			pass # TODO: proper handling
+
+		path = os.path.join(self.root, source, filename) # TODO: consider adding in date at some point
+		with open(path, 'wb') as f:
+			f.write(data)