add python 3 type hints

Author: alan412

backend/deploy.py

@@ -2,6 +2,7 @@
 import os
 import shutil
 import zipfile
+from typing import Dict, List, Tuple, Union
 
 # Third-party imports
 from flask import request
@@ -11,7 +12,7 @@
 from main import basedir
 
 class DeployResource(Resource):
-    def post(self):
+    def post(self) -> Union[Dict[str, Union[str, int, List[str]]], Tuple[Dict[str, str], int]]:
         """Upload and extract a zip file to the deploy directory"""
         if 'file' not in request.files:
             return {'error': 'No file provided'}, 400

backend/main.py

@@ -2,9 +2,10 @@
 import argparse
 import logging
 import os
+from typing import Tuple, Union
 
 # Third-party imports
-from flask import Flask, jsonify, request, send_from_directory
+from flask import Flask, jsonify, request, send_from_directory, Response
 from flask_restful import Api
 from flask_sqlalchemy import SQLAlchemy
 
@@ -23,14 +24,14 @@
 
 # Add CORS headers
 @app.after_request
-def after_request(response):
+def after_request(response: Response) -> Response:
     response.headers.add('Access-Control-Allow-Origin', '*')
     response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
     response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
     return response
 
 @app.before_request
-def handle_preflight():
+def handle_preflight() -> Union[Response, None]:
     if request.method == "OPTIONS":
         response = jsonify({'message': 'OK'})
         response.headers.add('Access-Control-Allow-Origin', '*')
@@ -53,7 +54,7 @@ def handle_preflight():
 
 # API health check endpoint to distinguish from static file serving
 @app.route('/api/status')
-def api_status():
+def api_status() -> Response:
     """Health check endpoint to identify backend server"""
     return jsonify({
         'status': 'ok',
@@ -65,7 +66,7 @@ def api_status():
 @app.route('/blocks/', defaults={'path': ''})
 @app.route('/blocks/<path:path>')
 @app.route('/blocks//<path:path>')  # Handle double slash
-def serve_frontend(path):
+def serve_frontend(path: str) -> Union[Response, Tuple[Response, int]]:
     """Serve static assets from dist/ directory with base path"""
     # Normalize path - remove leading slashes and clean up double slashes
     path = path.lstrip('/')
@@ -101,7 +102,7 @@ def serve_frontend(path):
 
 @app.route('/', defaults={'path': ''})
 @app.route('/<path:path>')
-def serve_static(path):
+def serve_static(path: str) -> Union[Response, Tuple[Response, int]]:
     """Serve static assets from dist/ directory"""
     # If path is empty, serve index.html
     if path == '':

backend/storage.py

@@ -1,3 +1,6 @@
+# Standard library imports
+from typing import Dict, List, Tuple, Union
+
 # Third-party imports
 from flask import request
 from flask_restful import Resource
@@ -11,7 +14,7 @@ class StorageEntry(db.Model):
     entry_key = db.Column(db.String(255), nullable=False, unique=True)
     entry_value = db.Column(db.Text, nullable=False)
 
-    def to_dict(self):
+    def to_dict(self) -> Dict[str, str]:
         return {
             'key': self.entry_key,
             'value': self.entry_value
@@ -22,15 +25,15 @@ class StorageFile(db.Model):
     file_path = db.Column(db.String(500), nullable=False, unique=True)
     file_content = db.Column(db.Text, nullable=False)
 
-    def to_dict(self):
+    def to_dict(self) -> Dict[str, str]:
         return {
             'path': self.file_path,
             'content': self.file_content
         }
 
 # Storage Resources for key-value and file operations
 class StorageEntryResource(Resource):
-    def get(self, entry_key):
+    def get(self, entry_key: str) -> Dict[str, str]:
         """Fetch entry value by key"""
         entry = StorageEntry.query.filter_by(entry_key=entry_key).first()
         if entry:
@@ -40,7 +43,7 @@ def get(self, entry_key):
             default_value = request.args.get('default', '')
             return {'value': default_value}
 
-    def post(self, entry_key):
+    def post(self, entry_key: str) -> Union[Dict[str, str], Tuple[Dict[str, str], int]]:
         """Save entry value"""
         data = request.get_json()
         if not data or 'value' not in data:
@@ -61,7 +64,7 @@ def post(self, entry_key):
             return {'error': 'Failed to save entry'}, 500
 
 class StorageResource(Resource):
-    def get(self, path):
+    def get(self, path: str) -> Union[Dict[str, Union[List[str], str]], Tuple[Dict[str, str], int]]:
         """Get file content or list directory based on path"""
         # Handle empty path as root directory
         if not path:
@@ -107,7 +110,7 @@ def get(self, path):
             else:
                 return {'error': 'File not found'}, 404
 
-    def post(self, path):
+    def post(self, path: str) -> Union[Dict[str, str], Tuple[Dict[str, str], int]]:
         """Save file content (only for files, not directories)"""
         if path.endswith('/'):
             return {'error': 'Cannot save content to a directory path'}, 400
@@ -130,7 +133,7 @@ def post(self, path):
             db.session.rollback()
             return {'error': 'Failed to save file'}, 500
 
-    def delete(self, path):
+    def delete(self, path: str) -> Union[Dict[str, str], Tuple[Dict[str, str], int]]:
         """Delete file or directory"""
         if path.endswith('/'):
             # Delete directory (all files starting with this path)
@@ -152,7 +155,7 @@ def delete(self, path):
             return {'error': 'Failed to delete'}, 500
 
 class StorageFileRenameResource(Resource):
-    def post(self):
+    def post(self) -> Union[Dict[str, str], Tuple[Dict[str, str], int]]:
         """Rename file or directory"""
         data = request.get_json()
         if not data or 'old_path' not in data or 'new_path' not in data:
@@ -182,7 +185,7 @@ def post(self):
             return {'error': 'Failed to rename'}, 500
 
 class StorageRootResource(Resource):
-    def get(self):
+    def get(self) -> Dict[str, List[str]]:
         """List all top-level files and directories"""
         files = StorageFile.query.all()
         # Extract top-level files and directories