class BlastImage(Resource):
def __init__(self):
if 'BLAST_IMAGE_DB_SERVICE_HOST' in os.environ:
self._db = PostgreSQL(os.environ['POSTGRESQL_USER'], \
os.environ['POSTGRESQL_PASSWORD'], \
os.environ['BLAST_IMAGE_DB_SERVICE_HOST'], \
os.environ['BLAST_IMAGE_DB_SERVICE_PORT'])
else:
self._db = PostgreSQL('user', 'password', 'localhost', '5432')
def get(self, tag):
items = []
for obj in self._db.get(tag):
items.append({'tag': obj['tag'], 'image': obj['image']})
return items
class BlastImage(Resource):
def __init__(self):
if 'IMAGE_DB_SERVICE_HOST' in os.environ:
self.db = PostgreSQL(os.getenv('POSTGRESQL_USER'), \
os.getenv('POSTGRESQL_PASSWORD'), \
os.getenv('IMAGE_DB_SERVICE_HOST'), \
os.getenv('IMAGE_DB_SERVICE_PORT'))
else:
self.db = PostgreSQL('user', 'password', 'localhost', '5432')
def get(self, tag):
items = []
try:
for obj in self.db.get(tag):
items.append({'tag': obj['tag'], 'image': obj['image']})
except Exception as e:
print(e, file=sys.stderr)
return items
