Skip to content
This repository was archived by the owner on Apr 16, 2023. It is now read-only.

Commit 735a9aa

Browse files
author
Patrick Kuehn
committed
Black code formating
- Rename File to temporary file
1 parent 2c7f764 commit 735a9aa

File tree

9 files changed

+20
-28
lines changed

9 files changed

+20
-28
lines changed

autofocus/predict/app/models/File.py renamed to autofocus/predict/app/filesystem/TemporaryFile.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55

66
UPLOAD_FOLDER = "/tmp/"
77

8-
class File:
8+
9+
class TemporaryFile:
910
"""
1011
Store a file and remove it upon destruction
1112

autofocus/predict/app/models/ZipArchive.py renamed to autofocus/predict/app/filesystem/ZipArchive.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
from zipfile import ZipFile
33

4-
from .File import File, UPLOAD_FOLDER
4+
from .TemporaryFile import TemporaryFile, UPLOAD_FOLDER
55
from ..validation.validation import allowed_file, ALLOWED_IMAGE_FILES
66

77

@@ -26,7 +26,7 @@ def __init__(self, file):
2626
file: Uploaded file from flask
2727
upload_folder: The folder to save the zip file
2828
"""
29-
self.file = File(file)
29+
self.file = TemporaryFile(file)
3030
self.zip = ZipFile(self.file.getPath())
3131

3232
def listFiles(self):
@@ -81,7 +81,7 @@ def extractAll(self, path=UPLOAD_FOLDER, members=None):
8181
self.zip.extractall(path, members)
8282
extractedFiles = {}
8383
for member in members:
84-
file = File()
84+
file = TemporaryFile()
8585
file.setPath(os.path.join(path, member))
8686
extractedFiles[member] = file
8787
return extractedFiles
File renamed without changes.

autofocus/predict/app/prediction/prediction.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def predict_multiple(files):
2424
predictions[key] = predict(files[key])
2525
return predictions
2626

27+
2728
def predict(file):
2829
"""
2930
Predict probabilities of single file
@@ -36,6 +37,7 @@ def predict(file):
3637
pred_classes, preds, probs = model.predict(image)
3738
return getProbabilities([prob.item() for prob in probs])
3839

40+
3941
def getProbabilities(probabilities):
4042
"""
4143
Return formated Probabilities
Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,21 @@
1-
import time
2-
31
from flask import Blueprint, jsonify, request
42

5-
from ..models.File import File
6-
from ..models.ZipArchive import ZipArchive
3+
from ..filesystem.TemporaryFile import TemporaryFile
74
from ..validation.predict import validate_predict_request
8-
from ..validation.predict_zip import validate_predict_zip_request
9-
from ..validation.validation import abort_with_errors
10-
from ..prediction.prediction import predict, predict_multiple
5+
from ..prediction.prediction import predict
116

127

138
predict_route = Blueprint("predict", __name__)
149

10+
1511
@predict_route.route("/predict", methods=["POST"])
1612
def classify_single():
1713
"""Classify a single image"""
1814
# Validate request
1915
validate_predict_request(request)
2016

2117
# Get File object
22-
file = File(request.files["file"])
18+
file = TemporaryFile(request.files["file"])
2319

2420
# Return ziped probabilities
2521
return jsonify(predict(file))

autofocus/predict/app/routes/predict_zip.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,14 @@
1-
import time
2-
31
from flask import Blueprint, jsonify, request
42

5-
from ..models.File import File
6-
from ..models.ZipArchive import ZipArchive
7-
from ..validation.predict import validate_predict_request
3+
from ..filesystem.ZipArchive import ZipArchive
84
from ..validation.predict_zip import validate_predict_zip_request
95
from ..validation.validation import abort_with_errors
10-
from ..prediction.prediction import predict, predict_multiple
6+
from ..prediction.prediction import predict_multiple
117

128

139
predict_zip_route = Blueprint("predict_zip", __name__)
1410

11+
1512
@predict_zip_route.route("/predict_zip", methods=["POST"])
1613
def classify_zip():
1714
"""Classify all images from a zip file"""
@@ -20,9 +17,7 @@ def classify_zip():
2017

2118
file = ZipArchive(request.files["file"])
2219
if not file.hasImages():
23-
error = {
24-
"file": "No image files detected in the zip file."
25-
}
20+
error = {"file": "No image files detected in the zip file."}
2621
abort_with_errors(error)
2722

2823
# Extract files

autofocus/predict/app/validation/predict.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,9 @@ def validate_predict_request(request):
1313
if not file:
1414
error["file"] = "No file given."
1515
elif not allowed_file(file.filename, ALLOWED_IMAGE_FILES):
16-
error[
17-
"file"
18-
] = "File type not allowed. File must be of type {allowed}".format(
16+
error["file"] = "File type not allowed. File must be of type {allowed}".format(
1917
allowed=ALLOWED_IMAGE_FILES
2018
)
2119

22-
if (error):
20+
if error:
2321
abort_with_errors(error)

autofocus/predict/app/validation/predict_zip.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from .validation import abort_with_errors, ALLOWED_ZIP_FILES, allowed_file
22

3+
34
def validate_predict_zip_request(request):
45
"""
56
Validate the given request
@@ -15,9 +16,7 @@ def validate_predict_zip_request(request):
1516
if not file:
1617
error["file"] = "No file given."
1718
elif not allowed_file(file.filename, ALLOWED_ZIP_FILES):
18-
error[
19-
"file"
20-
] = "File type not allowed. File must be of type {allowed}".format(
19+
error["file"] = "File type not allowed. File must be of type {allowed}".format(
2120
allowed=ALLOWED_ZIP_FILES
2221
)
2322

autofocus/predict/app/validation/validation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def abort_with_errors(error):
2020
)
2121
)
2222

23+
2324
def allowed_file(filename, allowed_extensions):
2425
"""
2526
Check for whether a filename is in the ALLOWED_EXTENSIONS

0 commit comments

Comments
 (0)