创建一个简单的数据库来存储映射关系
conn = sqlite3.connect('short_links.db')
cursor = conn.cursor()
cursor.execute('''CREATE TABLE IF NOT EXISTS links
(short_link TEXT PRIMARY KEY, long_link TEXT)''')
conn.commit()
conn.close()
@app.route('/shorten', methods=['POST'])
def shorten_link():
data = request.json
if 'long_link' not in data:
return jsonify({error: Missing long_link parameter}), 400