def add():
slj = models.Actor('Samuel L. Jackson')
bw = models.Actor('Bruce Willys')
m = models.Movie('Pulp Fiction', 5)
m.actors.append(slj)
m.actors.append(bw)
m.save()
mm = models.Movie('Breaking Bad', 5)
mm.actors.append(bw)
mm.save()
return "Added!"
def _db_movies(session):
movie1 = models.Movie(title='title1', description='description1', release_year=2018)
movie2 = models.Movie(title='title2', description='description2', release_year=2018)
movie3 = models.Movie(title='title3', description='description3', release_year=2020)
movie1.avg_rating = 6
movie2.avg_rating = 3
movie3.avg_rating = 7
session.add(movie1)
session.add(movie2)
session.add(movie3)
session.commit()
return movie1, movie2, movie3
def create_movie(session: Session, movie: schemas.MovieCreate) -> models.Movie:
db_movie = models.Movie(**movie.dict())
session.add(db_movie)
session.flush()
session.refresh(db_movie)
return db_movie
def add_movie():
movie = models.Movie.query.filter_by(name=request.json['name']).first()
if movie is None:
movie = models.Movie()
movie.init(request.json)
movie.api_owner = current_user.username
db.session.add(movie)
else:
movie.init(request.json)
movie.api_owner = current_user.username
db.session.commit()
return jsonify(movie.serialize)
def movie_add():
form = forms.Add_Movie()
path = app.config["UP_DIR"]
if form.validate_on_submit():
url = File(form.url.data, path)
logo = File(form.logo.data, path)
movie_obj = models.Movie(name=form.data['name'],
url=url.safe_filename_save(),
logo=logo.safe_filename_save(),
info=form.data['info'],
score=int(form.data['score']),
playnum=0,
area=form.data['area'],
length=form.data['length'],
category_id=form.data['tag'],
release_time=form.data['release_time'])
mysql_db.session.add(movie_obj)
mysql_db.session.commit()
AddLog.add_oplog('添加电影')
return redirect(url_for('admin.movie_list', page=1))
return render_template('admin/movie_add.html', form=form)
def movie_add():
form = MovieForm()
if form.validate_on_submit():
data = form.data
file_url = secure_filename(form.url.data.filename)
file_logo = secure_filename(form.logo.data.filename)
if not os.path.exists(app.config['UP_DIR']):
os.makedirs(app.config['UP_DIR'])
os.chmod(app.config['UP_DIR'], 'rw')
movie_count = app_models.Movie.query.filter_by(
title=data['title']).count()
if movie_count >= 1:
flash('片名已经存在!', 'err')
return redirect(url_for('admin.movie_add'))
url = change_filename(file_url)
logo = change_filename(file_logo)
form.url.data.save(app.config['UP_DIR'] + url)
form.logo.data.save(app.config['UP_DIR'] + logo)
movie = app_models.Movie(
title=data['title'],
info=data['info'],
logo='20180305140015b8d9dbef9ac74d15984f6c9fe8e8bb92.jpg',
star=int(data['star']),
playnum=0,
commentnum=0,
tag_id=int(data['tag_id']),
area=data['area'],
release_time=data['release_time'],
length=data['length'])
db.session.add(movie)
db.session.commit()
flash('添加电影成功', 'ok')
return redirect(url_for('admin.movie_add'))
return render_template('admin/movie_add.html', form=form)