This commit is contained in:
Joseph Schorr 2013-10-01 17:44:21 -04:00
commit b1ec63893d
3 changed files with 22 additions and 8 deletions

14
app.py
View file

@ -5,13 +5,19 @@ from flask import Flask
from flask.ext.principal import Principal from flask.ext.principal import Principal
from flask.ext.login import LoginManager from flask.ext.login import LoginManager
from flask.ext.mail import Mail from flask.ext.mail import Mail
from config import ProductionConfig, DebugConfig from config import ProductionConfig, DebugConfig, LocalHostedConfig
app = Flask(__name__) app = Flask(__name__)
is_prod = os.environ.get('STACK', '').strip().lower().startswith('prod') stack = os.environ.get('STACK', '').strip().lower()
config_object = ProductionConfig() if is_prod else DebugConfig() if stack.startswith('prod'):
app.config.from_object(config_object) config = ProductionConfig()
elif stack.startswith('localhosted'):
config = LocalHostedConfig()
else:
config = DebugConfig()
app.config.from_object(config)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

View file

@ -62,6 +62,14 @@ class DebugConfig(FlaskConfig, MailConfig, LocalStorage, SQLiteDB):
} }
class LocalHostedConfig(FlaskConfig, MailConfig, S3Storage, RDSMySQL):
REGISTRY_SERVER = 'localhost:5000'
LOGGING_CONFIG = {
'level': logging.DEBUG,
'format': LOG_FORMAT
}
class ProductionConfig(FlaskConfig, MailConfig, S3Storage, RDSMySQL): class ProductionConfig(FlaskConfig, MailConfig, S3Storage, RDSMySQL):
REGISTRY_SERVER = 'quay.io' REGISTRY_SERVER = 'quay.io'
LOGGING_CONFIG = { LOGGING_CONFIG = {

View file

@ -27,7 +27,7 @@ def get_tags(namespace, repository):
if permission.can() or model.repository_is_public(namespace, repository): if permission.can() or model.repository_is_public(namespace, repository):
tags = model.list_repository_tags(namespace, repository) tags = model.list_repository_tags(namespace, repository)
tag_map = {tag.name: tag.image.image_id for tag in tags} tag_map = {tag.name: tag.image.docker_image_id for tag in tags}
return jsonify(tag_map) return jsonify(tag_map)
abort(403) abort(403)
@ -42,7 +42,7 @@ def get_tag(namespace, repository, tag):
if permission.can() or model.repository_is_public(namespace, repository): if permission.can() or model.repository_is_public(namespace, repository):
tag_image = model.get_tag_image(namespace, repository, tag) tag_image = model.get_tag_image(namespace, repository, tag)
response = make_response(tag_image.image_id, 200) response = make_response(tag_image.docker_image_id, 200)
abort(403) abort(403)
@ -55,8 +55,8 @@ def put_tag(namespace, repository, tag):
permission = ModifyRepositoryPermission(namespace, repository) permission = ModifyRepositoryPermission(namespace, repository)
if permission.can(): if permission.can():
image_id = json.loads(request.data) docker_image_id = json.loads(request.data)
model.create_or_update_tag(namespace, repository, tag, image_id) model.create_or_update_tag(namespace, repository, tag, docker_image_id)
return make_response('Created', 200) return make_response('Created', 200)