一站式网上办事大厅

from flask import Flask, request, send_from_directory

import os

app = Flask(__name__)

UPLOAD_FOLDER = 'uploads'

app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 确保上传目录存在

os.makedirs(UPLOAD_FOLDER, exist_ok=True)

@app.route('/upload', methods=['POST'])

def upload_file():

if 'file' not in request.files:

return 'No file part', 400

file = request.files['file']

if file.filename == '':

return 'No selected file', 400

file.save(os.path.join(app.config['UPLOAD_FOLDER'], file.filename))

return 'File uploaded successfully'

@app.route('/download/')

def download_file(filename):

return send_from_directory(app.config['UPLOAD_FOLDER'], filename)

if __name__ == '__main__':

app.run(debug=True)

from flask import Flask, request, jsonify

import jwt

import datetime

app = Flask(__name__)

SECRET_KEY = 'your-secret-key'

def generate_token(user_id):

payload = {

'user_id': user_id,

'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)

}

token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')

return token

@app.route('/login', methods=['POST'])

def login():

username = request.json.get('username')

password = request.json.get('password')

# 这里应连接数据库验证用户信息

if username == 'admin' and password == '123456':

token = generate_token(1)

return jsonify({'token': token})

else:

return jsonify({'error': 'Invalid credentials'}), 401

@app.route('/upload', methods=['POST'])

def protected_upload():

token = request.headers.get('Authorization')

if not token:

return jsonify({'error': 'Missing token'}), 401

try:

payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])

user_id = payload['user_id']

# 继续处理上传逻辑...

return jsonify({'message': 'Upload successful for user {}'.format(user_id)})

except jwt.ExpiredSignatureError:

return jsonify({'error': 'Token expired'}), 401

except jwt.InvalidTokenError:

return jsonify({'error': 'Invalid token'}), 401

if __name__ == '__main__':

app.run(debug=True)

import boto3

from botocore.exceptions import NoCredentialsError

s3 = boto3.client('s3', aws_access_key_id='YOUR_ACCESS_KEY',

aws_secret_access_key='YOUR_SECRET_KEY')

def upload_to_s3(file_name, bucket, object_name=None):

if object_name is None:

object_name = file_name

try:

response = s3.upload_file(file_name, bucket, object_name)

return True

except FileNotFoundError:

print("The file was not found.")

return False

except NoCredentialsError:

print("Credentials not available.")

return False

def download_from_s3(bucket, object_name, file_name):

try:

s3.download_file(bucket, object_name, file_name)

return True

except Exception as e:

print(e)

return False