Merge pull request #31 from codeguru42/17-dynamic-file-name

17 dynamic file name

Author: codeguru42

go_capture/views.py

@@ -1,4 +1,5 @@
 import io
+from pathlib import Path
 
 import cv2
 import numpy as np
@@ -16,10 +17,17 @@ def capture(request):
     image = cv2.imdecode(image_data, cv2.IMREAD_COLOR)
     board = perspective.get_grid(image)
     black, white = find_stones.find_stones(board)
-    file = io.StringIO()
-    make_sgf(file, black, white)
-    file.seek(0)
-    return FileResponse(file.read(), status=201, as_attachment=True, content_type='application/x-go-sgf')
+    output_file = io.StringIO()
+    make_sgf(output_file, black, white)
+    output_file.seek(0)
+    filename = Path(image_file.name).stem
+    return FileResponse(
+        output_file.read(),
+        status=201,
+        filename=f'${filename}.sgf',
+        as_attachment=True,
+        content_type='application/x-go-sgf'
+    )
 
 
 @require_GET