def iepginput(d):
a_title = Title(name=d['program-title'])
b_waves = Waves(name=d['station'])
c_station = Station(name=d['station-name'])
d_comments = Comments(name=d['comment'])
op = timeset(d, start)
ed = timeset(d, end)
sg1 = d['genre-1'] * 16 + d['subgenre-1']
if d['genre-2'] == 0:
x = Program(title=a_title,
program_id=d['program-id'],
waves=b_waves,
station=c_station,
start=op,
end=ed,
gen_1=d['genre-1'],
sgn_1=sg1,
comments=d_comments)
x.save
else:
sg2 = d['genre-2'] * 16 + d['subgenre-2']
x = Program(title=a_title,
program_id=d['program-id'],
waves=b_waves,
station=c_station,
start=op,
end=ed,
gen_1=d['genre-1'],
sgn_1=sg1,
gen_2=d['genre-2'],
sgn_2=sg2,
comments=d_comments)
x.save
def addProgram(genre_id):
"""Add a program to specified genre.
Args:
genre_id (int): Primary key of specified genre.
Returns:
If user is not signed in, redirect to login page.
on GET: Page with form for user to submit new program details.
on POST: Redirect to read-only page showing program details.
"""
if request.method == 'GET':
genres = session.query(Genre).order_by('name').all()
genre = session.query(Genre).filter_by(id=genre_id).one()
return render_template('addProgram.html', genres=genres, genre=genre)
elif request.method == 'POST':
name = request.form.get('name')
yearBegan = request.form.get('yearBegan')
yearEnded = request.form.get('yearEnded')
description = request.form.get('description')
user_id = login_session['user_id']
numPrograms = session.query(Program).filter_by(genre_id=genre_id,
name=name).count()
if numPrograms > 0:
flash('The program you are attempting to add already exists.')
return redirect(request.path)
program = Program(name=name, yearBegan=yearBegan, yearEnded=yearEnded,
description=description, genre_id=genre_id,
user_id=user_id)
session.add(program)
session.commit()
flash("Program \"%s\" created." % program.name)
return redirect(url_for('showProgram', genre_id=genre_id,
program_id=program.id))
def create_programs(programs):
for prog in programs:
program_id = generate_random_uuid()
program = Program(program_id=program_id,
name=programs[prog]['name'],
description=programs[prog]['description'])
add_and_commit(program)
programs[prog]['program_id'] = program_id
def new_program(token):
try:
body = request.get_json()
if body is None:
abort(404)
new_division = body.get('division')
new_director = body.get('director')
new_program = Program(division=new_division, director=new_director)
new_program.insert()
new_program = Program.query.filter_by(id=new_program.id).first()
return jsonify({'success': True, 'programs': [new_program.long()]})
except:
abort(401)
def save_to_db(self):
"""Saves the current program to the database and destroys the display"""
name = self.name_entry.get()
if not name:
# TODO: do something here that warns them I guess
return
option = self.dropdown.get()
try:
program = self.program_dict[option]
except KeyError:
# TODO: warning thingy like above
return
to_add = Program(name=name,
process_name=program.name(),
location=program.exe())
session.add(to_add)
session.commit()
self.destroy()
def program_upload(request):
user = request.user
if request.method == 'POST':
if user.is_authenticated():
file = request.FILES.get('filename', '')
file_name = file.name
dest_dir = os.path.join(settings.USR_PROGRAM_ROOT, user.username)
if not os.path.exists(dest_dir):
os.makedirs(dest_dir)
# os.chmod(dest_dir, 0775)
full_path = os.path.join(dest_dir, file_name)
destination = open(full_path, "wb+")
for chunk in file.chunks():
destination.write(chunk)
destination.close()
language = request.POST['lang']
description = request.POST['description']
access = request.POST['access']
new_program = Program(owner = user, path = full_path, name = file_name, language = language, description = description, access = access)
new_program.save()
notice = "Congratulations! Your model has been successfully uploaded."
return render_to_response('program/success.html', RequestContext(request, {'program': new_program, 'notice': notice}))
else:
notice = "You must be logged in to upload programs"
form = UploadProgramForm()
return render_to_response('program/upload.html', RequestContext(request, {'form': form, 'notice': notice}))
else:
form = UploadProgramForm()
return render_to_response('program/upload.html', RequestContext(request, {'form': form}))
