gopy.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. #!/usr/bin/env python3
  2. import flask
  3. from flask import make_response, request, redirect, jsonify, render_template, url_for
  4. from flask_sqlalchemy import SQLAlchemy
  5. from flask_marshmallow import Marshmallow
  6. db = SQLAlchemy()
  7. app = flask.Flask(__name__)
  8. app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:m9Eb5OflQb@mysql.mysql.svc.cluster.local/gopy'
  9. db.init_app(app)
  10. ma = Marshmallow(app)
  11. class Link(db.Model):
  12. id = db.Column(db.Integer, primary_key=True)
  13. name = db.Column(db.String(255), unique=True)
  14. target = db.Column(db.String(255))
  15. def __repr__(self):
  16. return '<Link id=(id) name=(name) target=(target)>'.format(
  17. id=self.id, name=self.name, target=self.target
  18. )
  19. class LinkSchema(ma.Schema):
  20. class Meta:
  21. fields = ('id', 'name', 'target')
  22. def get_links():
  23. links = Link.query.all()
  24. schema = LinkSchema(many=True)
  25. link_json = schema.dump(links)
  26. print(link_json)
  27. return links
  28. #jsonify(link_json)
  29. @app.route('/<string:name>', strict_slashes=False, methods=['GET'])
  30. def redirect_to_link(name):
  31. try:
  32. link = Link.query.filter_by(name=name).first()
  33. except e:
  34. return redirect("/", code=302)
  35. if link is None:
  36. return redirect("/", code=302)
  37. return redirect(link.target, code=302)
  38. @app.post('/<string:link_name>/add', strict_slashes=False)
  39. def add_link(link_name):
  40. db.create_all()
  41. link = Link(name=link_name, target=request.form['target'])
  42. db.session.add(link)
  43. db.session.commit()
  44. return redirect("/", code=302)
  45. @app.route('/<string:link_name>/add', strict_slashes=False)
  46. def add_link_form(link_name):
  47. return render_template('add.html', link_name=link_name)
  48. @app.post('/<string:link_name>/delete', strict_slashes=False)
  49. def delete_link(link_name):
  50. link = Link.query.filter_by(name=link_name).first()
  51. db.session.delete(link)
  52. db.session.commit()
  53. return redirect(url_for('hello_world'))
  54. @app.route('/<string:link_name>/delete', strict_slashes=False)
  55. def delete_link_form(link_name):
  56. link = Link.query.filter_by(name=link_name).first()
  57. print(link)
  58. return render_template('delete.html', link=link)
  59. @app.route('/')
  60. def hello_world():
  61. links = get_links()
  62. print(links)
  63. return render_template('list.html', links=links)
  64. #@app.route('/links/<int:id>', methods=['GET'])
  65. if __name__ == "__main__":
  66. app.run(host='0.0.0.0')