code now follows the PEP8 standard

This commit is contained in:
Martin
2016-04-28 16:24:38 +02:00
parent df5869501a
commit ead66c2561
6 changed files with 67 additions and 56 deletions
+9 -7
View File
@@ -4,22 +4,24 @@ from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__) app = Flask(__name__)
app.config.from_object('config') app.config.from_object('config')
def on_404(e): def on_404(e):
return jsonify(dict(error='Not found')), 404 return jsonify(dict(error='Not found')), 404
app.errorhandler(404)(on_404) app.errorhandler(404)(on_404)
#SQLAlchemy # SQLAlchemy
db = SQLAlchemy(app) db = SQLAlchemy(app)
from api.views import designers from api.views import designers
app.register_blueprint(designers.mod) app.register_blueprint(designers.mod)
@app.route("/") @app.route("/")
def index(): def index():
data = { data = {
"info": { "info": {
"api_version": "0.0.1" "api_version": "0.0.1"
}
} }
} return jsonify(data)
return jsonify(data)
+11 -10
View File
@@ -1,15 +1,16 @@
from api import db from api import db
class Designer(db.Model): class Designer(db.Model):
__tablename__ = 'designers' __tablename__ = 'designers'
designerid = db.Column(db.Integer, primary_key=True) designerid = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False) name = db.Column(db.String(100), nullable=False)
path = db.Column(db.String(100), nullable=False) path = db.Column(db.String(100), nullable=False)
def to_json(self, nested=False): def to_json(self, nested=False):
return { return {
'designerid': self.designerid, 'designerid': self.designerid,
'name': self.name, 'name': self.name,
'path': self.path, 'path': self.path,
} }
+9 -8
View File
@@ -1,13 +1,14 @@
from api import db from api import db
class Locale(db.Model): class Locale(db.Model):
__tablename__ = 'locales' __tablename__ = 'locales'
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
value = db.Column(db.String(5), nullable=False) value = db.Column(db.String(5), nullable=False)
def to_json(self): def to_json(self):
return { return {
"id": self.id, "id": self.id,
"value": self.value "value": self.value
} }
+24 -20
View File
@@ -4,38 +4,42 @@ from api.models import Designer
mod = Blueprint('designers', __name__, url_prefix='/designers') mod = Blueprint('designers', __name__, url_prefix='/designers')
@mod.route("/", methods=["GET"]) @mod.route("/", methods=["GET"])
def list(): def list():
designers = Designer.query.all() designers = Designer.query.all()
return jsonify(designers=[d.to_json() for d in designers]) return jsonify(designers=[d.to_json() for d in designers])
@mod.route("/", methods=["POST"]) @mod.route("/", methods=["POST"])
def create(): def create():
data = request.json data = request.json
designer = Designer(**data) designer = Designer(**data)
db.session.add(designer) db.session.add(designer)
db.session.commit() db.session.commit()
return jsonify(designer.to_json()) return jsonify(designer.to_json())
@mod.route("/<int:id>", methods=["GET"]) @mod.route("/<int:id>", methods=["GET"])
def show(id): def show(id):
designer = Designer.query.get_or_404(id) designer = Designer.query.get_or_404(id)
return jsonify(designer.to_json()) return jsonify(designer.to_json())
@mod.route("/<id>", methods=["PUT"]) @mod.route("/<id>", methods=["PUT"])
def update(id): def update(id):
designer = Designer.query.get_or_404(id) designer = Designer.query.get_or_404(id)
data = request.json data = request.json
for k, v in data.items(): for k, v in data.items():
setattr(designer, k, v) setattr(designer, k, v)
db.session.add(designer) db.session.add(designer)
db.session.commit() db.session.commit()
return jsonify(designer.to_json()) return jsonify(designer.to_json())
@mod.route("/<id>", methods=["DELETE"]) @mod.route("/<id>", methods=["DELETE"])
def delete(id): def delete(id):
designer = Designer.query.get_or_404(id) designer = Designer.query.get_or_404(id)
db.session.delete(designer) db.session.delete(designer)
db.session.commit() db.session.commit()
return jsonify({"status": "ok"}) return jsonify({"status": "ok"})
+13 -10
View File
@@ -4,24 +4,27 @@ from api.models import Designer
manager = Manager(app) manager = Manager(app)
@manager.command @manager.command
def init_db(): def init_db():
db.create_all() db.create_all()
print("All tables created") print("All tables created")
@manager.command @manager.command
def drop_db(): def drop_db():
if prompt_bool("Are you sure you want to lose all your data?"): if prompt_bool("Are you sure you want to lose all your data?"):
db.drop_all() db.drop_all()
print("All tables dropped") print("All tables dropped")
@manager.command @manager.command
def seed_db(): def seed_db():
print("Adding sample data") print("Adding sample data")
designer = Designer(name="Martin", path='/martin') designer = Designer(name="Martin", path='/martin')
db.session.add(designer) db.session.add(designer)
db.session.commit() db.session.commit()
if __name__ == '__main__': if __name__ == '__main__':
manager.run() manager.run()