数据库

import psycopg2
from flask import Flask,request , jsonify

app = Flask(name)

RDS_CONFIG = {
"dbname" : "postgres",
"user" : "postgres",
"password" : "Wangyi178",
"host" : "database-1.cqbu442ikt9q.us-east-1.rds.amazonaws.com",
"port" : "5432"
}

@app.route('/health')
def health():
return "ok",200

@app.route('/add_book', methods=['POST'])
def add_book():
data = request.get_json()
data1 = data["book"]
conn = psycopg2.connect(RDS_CONFIG)
cur = conn.cursor()
cur.execute("INSERT INTO books (id,title,author,stock) VALUES (%s,%s,%s,%s)" , (data1["id"],data1["title"],data1["author"],data1["stock"]))
conn.commit()
cur.close()
conn.close()
return jsonify(
{
"message" : "success"
}
),200
@app.route('/get_book', methods=['GET'])
def get_book():
id = request.args.get("id")
conn = psycopg2.connect(
RDS_CONFIG)
cur = conn.cursor()
cur.execute("select id ,title , author,stock from books where id = %s" , (id,))
data1 = cur.fetchone()
cur.close()
conn.close()
id = data1[0]
title = data1[1]
author = data1[2]
stock = data1[3]
return jsonify(
{
"id":id , "title":title , "author":author , "stock":stock
}
),200

@app.route('/update_book', methods=['PATCH'])
def update_book():
id = request.args.get("id")
data = request.get_json()
data1 = data["changes"]
author = data1["author"]
stock = int(data1["stock"])
conn = psycopg2.connect(**RDS_CONFIG)
cur = conn.cursor()
cur.execute("update books set author = %s , stock = %s where id = %s",(author,stock,id))
conn.commit()
cur.close()
conn.close()
return jsonify(
{
"message" : "success"
}
),200

if name == "main":
app.run(host="0.0.0.0",port=80)

psql --host=database-1.cqbu442ikt9q.us-east-1.rds.amazonaws.com --port=5432 --username=postgres

posted @ 2026-11-10 17:02  tabtab--007  阅读(1)  评论(0)    收藏  举报