def load_orgs():
""" Load wildlife orgs from user.csv into database. """
print "Orgs"
Org.query.delete()
# Read u.user file and insert data
data = csv_to_tuples(user_csv_path)
for row in data:
user_id, name, show_address, address1, address2, city, state, zipcode, phone, desc, accept_volunteers = row[:
11]
org = Org(user_id=user_id,
name=name,
ein=None,
show_address=show_address,
address1=address1,
address2=address2,
city=city,
state=state,
zipcode=zipcode,
desc=desc,
phone=phone,
email=None,
website=None,
accept_animals=1,
accept_volunteers=accept_volunteers)
db.session.add(org)
db.session.commit()
def random_orgs():
org_name = forgery_py.forgery.basic.text(length=None,
at_least=3,
at_most=15,
lowercase=True,
uppercase=True,
digits=False,
spaces=True,
punctuation=False)
password = forgery_py.forgery.basic.text(length=None,
at_least=3,
at_most=15,
lowercase=True,
uppercase=True,
digits=False,
spaces=False,
punctuation=False)
org_url = forgery_py.forgery.internet.top_level_domain()
an_org = Org(org_name=org_name, password=password, org_url=org_url)
db.session.add(an_org)
for i in range(4):
random_orgs()
def create_org_with_cause_id(org_name, cause_id, mission, web_url, tagline):
"""Create and return a new organization."""
org = Org(org_name=org_name,
cause_id=cause_id,
mission=mission,
web_url=web_url,
tagline=tagline)
db.session.add(org)
db.session.commit()
return org
def sample_data():
"""Create some sample data."""
# In case this is run more than once, empty out existing data
Cause.query.delete()
Org.query.delete()
# Add sample Organizations
womens_services = Cause(cause_name='Womens Services')
swap = Org(
org_name='DIVAS',
cause=womens_services,
mission=
'To support women escaping domestic abuse and homelessness and provide them with marketable skills'
)
db.session.add_all([womens_services, swap])
db.session.commit()
def user_added():
"""New account form submission."""
print request.form.get
# Basic user fields:
username = request.form.get("username")
email = request.form.get("email")
password = request.form.get("password")
account_made = datetime.now()
user = User(email=email,
username=username,
password=password,
account_made=account_made)
db.session.add(user)
db.session.flush()
db.session.commit()
# Automatically log the new user in
session['user_id'] = user.id
flash("Welcome to WildAlly, {}!".format(username))
# Check if the user is affiliated with an org:
is_org = request.form.get("is_org")
# If so, insert data into orgs and pickups:
if is_org == 'yes':
# Wildlife rehabilitator org fields:
name = request.form.get("org_name")
ein = request.form.get("ein")
show_address = request.form.get("show_address")
address1 = request.form.get("address1")
address2 = request.form.get("address2")
city = request.form.get("city")
state = request.form.get("state")
zipcode = request.form.get("zipcode")
desc = request.form.get("desc")
phone = request.form.get("phone")
org_email = request.form.get("org_email")
website = request.form.get("website")
org = Org(user_id=user.id,
name=name,
ein=ein,
show_address=show_address,
address1=address1,
address2=address2,
city=city,
state=state,
zipcode=zipcode,
desc=desc,
phone=phone,
email=org_email,
website=website)
db.session.add(org)
db.session.flush()
# Create geocode and insert coordinates into pickups:
org = Org.query.get(org.id)
if show_address == '1':
address = [address1, address2, city, state, zipcode]
else:
address = [city, state, zipcode]
coords = org.make_geocode()
pickup = Pickup(org_id=org.id, latitude=coords[0], longitude=coords[1])
db.session.add(pickup)
db.session.commit()
return redirect('/settings')
