基于研究生信息管理系统的航天数据平台设计与实现

from flask import Flask, request, jsonify

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://username:password@localhost/grad_system'

db = SQLAlchemy(app)

学工管理

class User(db.Model):

id = db.Column(db.Integer, primary_key=True)

name = db.Column(db.String(50))

email = db.Column(db.String(50))

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

def add_user():

data = request.get_json()

new_user = User(name=data['name'], email=data['email'])

db.session.add(new_user)

db.session.commit()

return jsonify({'message': 'User added'}), 201

]]>

class Satellite(db.Model):

id = db.Column(db.Integer, primary_key=True)

name = db.Column(db.String(50))

orbit_data = db.Column(db.Text)

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

def add_satellite():

data = request.get_json()

new_satellite = Satellite(name=data['name'], orbit_data=data['orbit_data'])

db.session.add(new_satellite)

db.session.commit()

return jsonify({'message': 'Satellite added'}), 201

]]>

import unittest

class TestAPI(unittest.TestCase):

def setUp(self):

self.app = app.test_client()

def test_add_user(self):

response = self.app.post('/users', json={'name': 'Alice', 'email': 'alice@example.com'})

self.assertEqual(response.status_code, 201)

if __name__ == '__main__':

unittest.main()

]]>