def upload_file(): title = request.form['song'] artist = request.form['artist'] picture_file = request.files['file'] photo = Photo() photo.file.put(picture_file) photo.save() track = Track.get_from_spotify(artist, title) if track is None: return "No track found!" track.save() submission = Submission(photo=photo, track=track) submission.save() return redirect('/submissions')
return string.lower().strip() if __name__ == '__main__': for track_photo_name in sorted(os.listdir(PICTURES_FOLDER)): path = os.path.join(PICTURES_FOLDER, track_photo_name) # Photo object photo = Photo() track_photo = open(path, 'rb') photo.file.put(track_photo) photo.save() # Track object artist, title = track_photo_name.split('.')[0].split('_') # Trying to find a previously created track object track = None for t in Track.objects: if clean(t.name) == clean(title) and clean(t.artist_name) == clean(title): track = t break if track is None: track = Track.get_from_spotify(artist, title) if track is None: print "! Couldn't get info for '{} - {}' ; skipping.".format(artist, title) continue track.save() # Submission submission = Submission(photo=photo, track=track) submission.save() print "* '{} - {}' saved in the db".format(artist, title)