Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 28 additions & 1 deletion app/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,34 @@
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
import os
from dotenv import load_dotenv

db = SQLAlchemy()
migrate = Migrate()

def create_app(test_config=None):
load_dotenv()


def create_app(testing=None):
app = Flask(__name__)

app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
if not testing:
app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("RENDER_DATABASE_URI")
else:
app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("SQLALCHEMY_TEST_DATABASE_URI")

from .models.moon import Moon
from .models.planet import Planet

db.init_app(app)
migrate.init_app(app, db)

from .routes.routes import solar_system_planet
app.register_blueprint(solar_system_planet)

from .routes.routes import solar_system_moon
app.register_blueprint(solar_system_moon)

return app
24 changes: 24 additions & 0 deletions app/models/moon.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
from app import db

class Moon(db.Model):
id = db.Column(db.Integer, primary_key = True, autoincrement = True)
name = db.Column(db.String)
radius = db.Column(db.Integer)
planet_id = db.Column(db.Integer)

def to_dict(self):
return {
"id": self.id,
"name": self.name,
"radius": self.radius,
"planet_id": self.planet_id
}

@classmethod
def from_dict(cls, moon_data):
return cls(
name = moon_data["name"],
radius = moon_data["radius"],
planet_id = moon_data["planet_id"],
)

24 changes: 24 additions & 0 deletions app/models/planet.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
from app import db

class Planet(db.Model):
id = db.Column(db.Integer, primary_key = True, autoincrement = True)
name = db.Column(db.String)
radius = db.Column(db.Integer)
description = db.Column(db.Text)

def to_dict(self):
return {
"id": self.id,
"name": self.name,
"radius": self.radius,
"description": self.description
}

@classmethod
def from_dict(cls, planet_data):
return cls(
name = planet_data["name"],
radius = planet_data["radius"],
description = planet_data["description"],
)

2 changes: 0 additions & 2 deletions app/routes.py

This file was deleted.

148 changes: 148 additions & 0 deletions app/routes/routes.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,148 @@
from flask import Blueprint, jsonify, request, make_response, abort
from app import db
from app.models.moon import Moon
from app.models.planet import Planet

solar_system_planet = Blueprint("solar_system_planet", __name__, url_prefix="/solar_system/planet")

@solar_system_planet.route("", methods=["POST"])
def add_planet():
request_body = request.get_json()
new_planet = Planet.from_dict(request_body)

db.session.add(new_planet)
db.session.commit()

return make_response({"message": f"Planet {new_planet.name} has been added, with the id: {new_planet.id}"}, 201)

@solar_system_planet.route("", methods=["GET"])
def get_planets():
return_list = []
name_query = request.args.get("name")
radius_query = request.args.get("radius_min")
if not name_query and not radius_query:
all_planets = Planet.query.all()
elif name_query and radius_query:
all_planets = Planet.query.filter(Planet.name==name_query, Planet.radius>=radius_query)
elif radius_query:
all_planets = Planet.query.filter(Planet.radius>=radius_query)
elif name_query:
all_planets = Planet.query.filter(Planet.name==name_query)
for planet in all_planets:
return_list.append({
"id": planet.id,
"name": planet.name,
"radius": planet.radius,
"description": planet.description
})
return jsonify(return_list), 200

@solar_system_planet.route("/<planet_id>", methods=["GET"])
def get_planet_by_id(planet_id):
planet = verify_item_id(Planet, planet_id)
return {
"id": planet.id,
"name": planet.name,
"radius": planet.radius,
"description": planet.description
}

@solar_system_planet.route("/<planet_id>", methods=["DELETE"])
def delete_planet(planet_id):
planet = verify_item_id(Planet, planet_id)

db.session.delete(planet)
db.session.commit()

return make_response({"message": f"Planet {planet.id} has been deleted"}, 200)

@solar_system_planet.route("/<planet_id>", methods=["PUT"])
def update(planet_id):
planet = verify_item_id(Planet, planet_id)
request_body = request.get_json()

planet.name = request_body["name"]
planet.radius = request_body["radius"]
planet.description = request_body["description"]

db.session.commit()

return make_response({"message": f"Planet {planet.id} has been updated"}, 200)

solar_system_moon = Blueprint("solar_system_moon", __name__, url_prefix="/solar_system/moon")

@solar_system_moon.route("", methods=["POST"])
def add_moon():
request_body = request.get_json()
new_moon = Moon.from_dict(request_body)

db.session.add(new_moon)
db.session.commit()

return make_response({"message": f"Moon {new_moon.name} has been added, with the id: {new_moon.id}"}, 201)

@solar_system_moon.route("", methods=["GET"])
def get_moons():
return_list = []
name_query = request.args.get("name")
radius_query = request.args.get("radius_min")
if not name_query and not radius_query:
all_moons = Moon.query.all()
elif name_query and radius_query:
all_moons = Moon.query.filter(Moon.name==name_query, Moon.radius>=radius_query)
elif radius_query:
all_moons = Moon.query.filter(Moon.radius>=radius_query)
elif name_query:
all_moons = Moon.query.filter(Moon.name==name_query)
for moon in all_moons:
return_list.append({
"id": moon.id,
"name": moon.name,
"radius": moon.radius,
"planet name": moon.planet_id
})
return jsonify(return_list), 200

@solar_system_moon.route("/<moon_id>", methods=["GET"])
def get_moon_by_id(moon_id):
moon = verify_item_id(Moon, moon_id)
return {
"id": moon.id,
"name": moon.name,
"radius": moon.radius,
"planet name": moon.planet_id
}

@solar_system_moon.route("/<moon_id>", methods=["DELETE"])
def delete_planet(moon_id):
moon = verify_item_id(Moon, moon_id)

db.session.delete(moon)
db.session.commit()

return make_response({"message": f"Moon {moon.id} has been deleted"}, 200)

@solar_system_moon.route("/<moon_id>", methods=["PUT"])
def update(moon_id):
moon = verify_item_id(Moon, moon_id)
request_body = request.get_json()

moon.name = request_body["name"]
moon.radius = request_body["radius"]
moon.planet_id = request_body["planet_id"]

db.session.commit()

return make_response({"message": f"Moon {moon.id} has been updated"}, 200)

def verify_item_id(model, item_id):
try:
item_id = int(item_id)
except ValueError:
abort(make_response({"message": f"Moon {item_id} is invalid"}, 400))
all_items = model.query.all()
for item in all_items:
if item.id == item_id:
return item
abort(make_response({"message": f"Moon {item_id} is not found"}, 404))

1 change: 1 addition & 0 deletions migrations/README
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Generic single-database configuration.
45 changes: 45 additions & 0 deletions migrations/alembic.ini
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
# A generic, single database configuration.

[alembic]
# template used to generate migration files
# file_template = %%(rev)s_%%(slug)s

# set to 'true' to run the environment during
# the 'revision' command, regardless of autogenerate
# revision_environment = false


# Logging configuration
[loggers]
keys = root,sqlalchemy,alembic

[handlers]
keys = console

[formatters]
keys = generic

[logger_root]
level = WARN
handlers = console
qualname =

[logger_sqlalchemy]
level = WARN
handlers =
qualname = sqlalchemy.engine

[logger_alembic]
level = INFO
handlers =
qualname = alembic

[handler_console]
class = StreamHandler
args = (sys.stderr,)
level = NOTSET
formatter = generic

[formatter_generic]
format = %(levelname)-5.5s [%(name)s] %(message)s
datefmt = %H:%M:%S
96 changes: 96 additions & 0 deletions migrations/env.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
from __future__ import with_statement

import logging
from logging.config import fileConfig

from sqlalchemy import engine_from_config
from sqlalchemy import pool
from flask import current_app

from alembic import context

# this is the Alembic Config object, which provides
# access to the values within the .ini file in use.
config = context.config

# Interpret the config file for Python logging.
# This line sets up loggers basically.
fileConfig(config.config_file_name)
logger = logging.getLogger('alembic.env')

# add your model's MetaData object here
# for 'autogenerate' support
# from myapp import mymodel
# target_metadata = mymodel.Base.metadata
config.set_main_option(
'sqlalchemy.url',
str(current_app.extensions['migrate'].db.engine.url).replace('%', '%%'))
target_metadata = current_app.extensions['migrate'].db.metadata

# other values from the config, defined by the needs of env.py,
# can be acquired:
# my_important_option = config.get_main_option("my_important_option")
# ... etc.


def run_migrations_offline():
"""Run migrations in 'offline' mode.

This configures the context with just a URL
and not an Engine, though an Engine is acceptable
here as well. By skipping the Engine creation
we don't even need a DBAPI to be available.

Calls to context.execute() here emit the given string to the
script output.

"""
url = config.get_main_option("sqlalchemy.url")
context.configure(
url=url, target_metadata=target_metadata, literal_binds=True
)

with context.begin_transaction():
context.run_migrations()


def run_migrations_online():
"""Run migrations in 'online' mode.

In this scenario we need to create an Engine
and associate a connection with the context.

"""

# this callback is used to prevent an auto-migration from being generated
# when there are no changes to the schema
# reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html
def process_revision_directives(context, revision, directives):
if getattr(config.cmd_opts, 'autogenerate', False):
script = directives[0]
if script.upgrade_ops.is_empty():
directives[:] = []
logger.info('No changes in schema detected.')

connectable = engine_from_config(
config.get_section(config.config_ini_section),
prefix='sqlalchemy.',
poolclass=pool.NullPool,
)

with connectable.connect() as connection:
context.configure(
connection=connection,
target_metadata=target_metadata,
process_revision_directives=process_revision_directives,
**current_app.extensions['migrate'].configure_args
)

with context.begin_transaction():
context.run_migrations()


if context.is_offline_mode():
run_migrations_offline()
else:
run_migrations_online()
Loading