Work with Python 3.

Author: mzhang28

.gitignore

@@ -3,3 +3,4 @@ __pycache__
 .data
 .cache
 .coverage
+uploads

.travis.yml

@@ -1,7 +1,7 @@
 language: python
 sudo: required
 python:
-  - "2.7"
+  - "3.5"
 install:
   - pip install -r requirements.txt
 before_script:

app.py

@@ -6,10 +6,13 @@
 app = Flask(__name__)
 app.config["UPLOAD_FOLDER"] = os.getenv("UPLOAD_FOLDER", "/usr/share/nginx/html")
 
+if not os.path.exists(app.config["UPLOAD_FOLDER"]):
+    os.makedirs(app.config["UPLOAD_FOLDER"])
+
 
 @app.route("/")
 def index():
-    return "Hello."
+    return "You shouldn't be here."
 
 
 @app.route("/save", methods=["POST"])
@@ -30,4 +33,5 @@ def save():
 
 
 if __name__ == "__main__":
-    app.run(use_debugger=True, use_reloader=True, port=8000, host="0.0.0.0")
+    port = int(os.getenv("PORT", "8000"))
+    app.run(use_debugger=True, use_reloader=True, port=port, host="0.0.0.0")

tests/test_general.py

@@ -1,4 +1,4 @@
-from cStringIO import StringIO
+from io import BytesIO as StringIO
 import os
 
 
@@ -12,27 +12,26 @@ def test_no_file_uploaded(self, client):
         assert r.status_code == 400
 
     def test_missing_filename(self, client):
-        file = StringIO("the quick brown fox jumps over the lazy dog")
+        file = StringIO(b"the quick brown fox jumps over the lazy dog")
         r = client.post("/save", data=dict(file=(file, "")))
         assert r.status_code == 400
 
     def test_save_prefix(self, client):
         prefix = "hello"
-        file = StringIO("the quick brown fox jumps over the lazy dog")
-        url = client.post(
-            "/save", data=dict(file=(file, "hello.txt"), prefix=prefix)).data
-        assert url[:len(prefix)] == prefix
+        file = StringIO(b"the quick brown fox jumps over the lazy dog")
+        url = client.post("/save", data=dict(file=(file, "hello.txt"), prefix=prefix)).data
+        assert url[:len(prefix)].decode() == prefix
 
     def test_save_suffix(self, client):
         suffix = "hello.txt"
-        file = StringIO("the quick brown fox jumps over the lazy dog")
-        url = client.post(
-            "/save", data=dict(file=(file, "hello.txt"), suffix=suffix)).data
-        assert url[-len(suffix):] == suffix
+        file = StringIO(b"the quick brown fox jumps over the lazy dog")
+        url = client.post("/save", data=dict(file=(file, "hello.txt"), suffix=suffix)).data
+        assert url[-len(suffix):].decode() == suffix
 
     def test_retrieved(self, app, client):
-        data = "the quick brown fox jumps over the lazy dog"
+        data = b"the quick brown fox jumps over the lazy dog"
         file = StringIO(data)
         url = client.post("/save", data=dict(file=(file, "hello.txt"))).data
-        with open(os.path.join(app.config["UPLOAD_FOLDER"], url), "r") as f:
-            assert f.read() == data
+        path = os.path.join(app.config["UPLOAD_FOLDER"], url.decode())
+        with open(path, "r") as f:
+            assert f.read() == data.decode()