def add_collection(user, auth):
user = get_user_by_name(session, user)
if not user:
return error_user_not_found()
if not is_user_logged_in(session, auth, user.display_name):
return error_request_denied()
body = bottle.request.json
name_missing = "name" not in body
description_missing = "description" not in body
if name_missing or description_missing:
bottle.response.status = 400
return {"error": "Missing required parameters."}
name = body["name"]
description = body["description"]
already_exists = get_collection_by_name(session, name)
if already_exists:
bottle.response.status = 400
return {"error": "There is already a collection with that name."}
new_collection = Collection(user=user, name=name, description=description)
session.add(new_collection)
session.commit()
return {"id": new_collection.id}
def add_new(self, name, type):
new_collection = Collection(name=name, type=type)
db_session.add(new_collection)
db_session.commit()
return new_collection
def setUp(self):
DB.drop_all()
DB.create_all()
collect = Collection('1', 'National Museum of Art',
'http://www.nga.gov', 'North America',
'Institution')
DB.session.add(collect)
DB.session.commit()
def create_collection(user_collection, name):
"""Creates a collection of podcasts for a user."""
collect = Collection(user_collection=user_collection, name=name)
db.session.add(collect)
db.session.commit()
return collect
def create_collection(user, collection_type, lendable=False, private=False):
collection = Collection(user=user,
collection_type=collection_type,
lendable=lendable,
private=private)
db.session.add(collection)
db.session.commit()
return collection
def create_collection(id):
form = forms.NewCollectionForm(request.form)
if not form.validate():
flash("Error, all fields are required")
else:
collection = Collection(title=form.title.data, description=form.description.data, user_id=current_user.id)
model.session.add(collection)
model.session.commit()
return redirect(url_for("view_user", id=current_user.id))
def create_collection(user_id, notes, date_saved=datetime.today()):
"""Create and return a new collection."""
new_collection = Collection(user_id=user_id,
notes=notes,
date_saved=date_saved)
db.session.add(new_collection)
db.session.commit()
return new_collection
def pet_star():
info = {'result': 0, 'msg': "成功", 'data': None}
json_data = json.loads(request.get_data())
phone = json_data['phone']
pet_code = json_data['PetCode']
star = json_data['star']
if star is True:
c = Collection(phone=phone, pet_code=pet_code)
db.session.add(c)
db.session.commit()
elif star is False:
c = Collection(phone=phone, pet_code=pet_code)
db.session.delete(c)
db.session.commit()
else:
info['result'] = 1
info['msg'] = '失败'
info['data'] = '操作失败'
return jsonify(info)
def test_addition_deletion(self):
"""
Test adding and deleting
"""
collect = Collection('2', 'Kimbell Museum of Art',
'https://www.kimbellart.org', 'North America',
'Institution')
DB.session.add(collect)
DB.session.commit()
self.assertEqual(len(Collection.query.all()), 2)
Collection.query.filter_by(
institution='Kimbell Museum of Art').delete()
DB.session.commit()
self.assertEqual(len(Collection.query.all()), 1)
def preload_tables(self, db):
base_path = os.path.split(__file__)[0]
#self.resource_path = os.path.join(base_path, "files", "oneclick")
path = os.path.join(base_path, "example.json")
data = open(path).read()
dict_data = json.loads(data)
collections = dict_data['collections']
coll_1 = Collection()
coll_1.name = collections[0]['name']
coll_1.curator = collections[0]['curator']
display_items = collections[0]['display_items']
db.commit()
print "hi"
def upload():
""" DOCS: http://zabana.me/notes/upload-files-amazon-s3-flask.html """
files = request.files.getlist('user_file')
user = User.query.filter(User.username == session['username']).first()
# Check if there's a 'user_file' key
if len(files) == 0:
flash('No files were selected. Please select file(s) to upload.')
return redirect('/new_collection')
# Create a new collection instance and commit to database
uuid_string = str(uuid.uuid4())
new_collection = Collection(user=user, uuid=uuid_string)
db.session.add(new_collection)
db.session.commit()
# Upload photos to s3
try:
upload_files(files, new_collection.id)
except:
flash('Failed to upload files. Please try again. ')
return redirect('/new_collection')
# Faces indexing and matching
if create_rekognition_collection(new_collection.id) == True:
try:
for photo in new_collection.photos:
index_faces(new_collection.id, photo.s3_key, photo.id)
process_faces(new_collection.id)
except:
flash('Failed to process files. Please try again. ')
return redirect('/new_collection')
# Update the database with the time finish processing the collection
new_collection.time_processed = datetime.utcnow()
db.session.commit()
flash('Your collection of pictures was processed successfully!')
# Delete the collection when it's done
delete_rekognition_collection(new_collection.id)
return redirect(f'/collections/{new_collection.id}')
def createCollection(self,
title,
creator,
description,
owner,
workUUIDs=[],
editionIDs=[]):
newCollection = Collection(uuid=uuid4(),
title=title,
creator=creator,
description=description,
owner=owner)
collectionEditions = []
if len(workUUIDs) > 0:
collectionWorks = self.session.query(Work)\
.join(Work.editions)\
.filter(Work.uuid.in_(workUUIDs))\
.all()
for work in collectionWorks:
editions = list(
sorted([ed for ed in work.editions],
key=lambda x: x.publication_date
if x.publication_date else date.today()))
for edition in editions:
if len(edition.items) > 0:
collectionEditions.append(edition)
break
if len(editionIDs) > 0:
collectionEditions.extend(
self.session.query(Edition).filter(
Edition.id.in_(editionIDs)).all())
newCollection.editions = collectionEditions
self.session.add(newCollection)
return newCollection
def add_new_collection():
"""Add a new collection to user from profile."""
user_email = session['email']
user = User.query.filter_by(email=user_email).first()
new_title = request.form['title']
existing_collection = Collection.query.filter_by(title=new_title,
user_id=user.id).first()
if existing_collection == None:
new_collection = Collection(title=new_title, user_id=user.id)
db.session.add(new_collection)
db.session.commit()
flash('Successfully added new collection: %s' % new_collection.title)
return redirect('/profile')
else:
flash('Collection with this title already exists')
return redirect('/profile')
def load_collections():
print "Adding User Collections"
user_id = User.query.filter_by(
email=os.environ['SAMPLE_EMAIL']).first().user_id
with open('data/recipe_files/past_meals.csv') as file:
for line in file:
line = line.rstrip().split(',')
col_data = Collection(user_id=user_id,
assigned_date=line[2].lstrip().rstrip(),
set_number=line[3].lstrip().rstrip(),
set_day=line[4].lstrip().rstrip(),
meal_type=line[5].lstrip().rstrip(),
recipe_id=line[6].lstrip().rstrip())
db.session.add(col_data)
db.session.commit()
def display_current_meal_plan():
""" If no meal plan currently exists, creates one. If meal plan already
exits, displays the current week's meal plan
Collection attribute 'set_number' is a concat of current year, week of year,
i.e. YYYYWW
Collection attribute 'set_day' is a single digit indicating the number of the
day of the week to which it corresponds, i.e. 1 == Monday, 2 == Tuesday, etc. """
week = datetime.date.today().strftime("%W")
thisweek = Week.thisweek().week
year = datetime.date.today().strftime("%Y")
start_date = datetime.datetime.strptime(year + "-W" + str(thisweek) + "-1",
"%Y-W%W-%w")
start_day = start_date.strftime("%b %d, %Y")
set_number = year + week
user_id = session['user_id']
# If Collection does not already exist for the current week, create one
if len(
Collection.query.filter(Collection.set_number == set_number,
Collection.user_id == user_id).all()) == 0:
user_prefs = []
prefs = db.session.query(UserPreference.category_id).filter(
UserPreference.user_id == user_id).all()
for pref in prefs:
user_prefs.append(pref[0])
if len(user_prefs) > 0:
breakfast = Category.query.filter(
Category.category_name.in_(breakfast_list)).all()
lunch = Category.query.filter(
Category.category_id.in_(user_prefs)).all()
dinner = Category.query.filter(
Category.category_id.in_(user_prefs)).all()
else:
breakfast = Category.query.filter(
Category.category_name.in_(breakfast_list)).all()
lunch = Category.query.filter(
Category.category_name.in_(lunch_list)).all()
dinner = Category.query.filter(
Category.category_name.in_(dinner_list)).all()
weekly_breakfasts = meal_plan.make_meal_set(breakfast, user_prefs)
weekly_lunches = meal_plan.make_meal_set(lunch, user_prefs)
weekly_dinners = meal_plan.make_meal_set(dinner, user_prefs)
# Create Collection object from this week's selected breakfast recipes
day_num = 0
for item in weekly_breakfasts:
day_num += 1
breakfast_recipe = Collection(user_id=session['user_id'],
assigned_date=start_day,
set_number=set_number,
set_day=day_num,
meal_type="breakfast",
recipe_id=item.recipe_id)
db.session.add(breakfast_recipe)
# Create Collection object from this week's selected lunch recipes
day_num = 0
for item in weekly_lunches:
day_num += 1
lunch_recipe = Collection(user_id=session['user_id'],
assigned_date=start_day,
set_number=set_number,
set_day=day_num,
meal_type="lunch",
recipe_id=item.recipe_id)
db.session.add(lunch_recipe)
# Create Collection object from this week's selected dinner recipes
day_num = 0
for item in weekly_dinners:
day_num += 1
dinner_recipe = Collection(user_id=session['user_id'],
assigned_date=start_day,
set_number=set_number,
set_day=day_num,
meal_type="dinner",
recipe_id=item.recipe_id)
db.session.add(dinner_recipe)
db.session.commit()
# Pull this week's Collection objects to pass into the HTML template &
# render in the meal plan page
breakfasts = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "breakfast").order_by(
Collection.set_day).all()
lunches = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "lunch").order_by(
Collection.set_day).all()
dinners = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "dinner").order_by(
Collection.set_day).all()
# Get shopping list using Ingredients module
final_ingredients = ingredients.get_shopping_list(
breakfasts, lunches, dinners)
return render_template("meal_plan.html",
now=week,
breakfasts=breakfasts,
lunches=lunches,
dinners=dinners,
ingredients=final_ingredients,
start_day=start_day)
# If Collection objects already exist for this week's meal plan, query them
# and return them with the HTML template
else:
breakfasts = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "breakfast").order_by(
Collection.set_day).all()
lunches = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "lunch").order_by(
Collection.set_day).all()
dinners = Collection.query.filter(
Collection.user_id == user_id, Collection.set_number == set_number,
Collection.meal_type == "dinner").order_by(
Collection.set_day).all()
# Generate shopping list using Ingredients module
final_ingredients = ingredients.get_shopping_list(
breakfasts, lunches, dinners)
return render_template("meal_plan.html",
now=week,
breakfasts=breakfasts,
lunches=lunches,
dinners=dinners,
ingredients=final_ingredients,
start_day=start_day)
display_name='Paul Bob',
password_hash=bcrypt.hashpw('PaulPass'.encode(), bcrypt.gensalt())))
session.add(
User(email='*****@*****.**',
display_name='John Bob',
password_hash=bcrypt.hashpw('JohnPass'.encode(), bcrypt.gensalt())))
RueBob = User(email='*****@*****.**',
display_name='Rue Bob',
password_hash=bcrypt.hashpw('RuePass'.encode(),
bcrypt.gensalt()))
session.add(RueBob)
fossil_collection = Collection(user=RueBob,
name="Fossils",
description="Some fossils")
butterfly_collection = Collection(user=RueBob,
name="Butterflies",
description="Some butterflies")
session.add(butterfly_collection)
session.add(fossil_collection)
fossil = Item(collection=fossil_collection,
name="A fossil",
description="A fossil for the collection")
session.add(fossil)
butterfly = Item(collection=butterfly_collection,