def dat2base():
'''transefer dat to database'''
path = '../ml-latest-small/movies.csv'
count = 0
fp = open(path, 'r')
fp.readline()
regex = re.compile("(\d+),\"?(.+)\"?,(.+)")
for line in fp.readlines():
ilt = regex.search(line)
if ilt.group(2).endswith("\""):
movie = Movie(movie_id=ilt.group(1),
movie_name=ilt.group(2)[:-2],
movie_genres=ilt.group(3))
else:
movie = Movie(movie_id=ilt.group(1),
movie_name=ilt.group(2),
movie_genres=ilt.group(3))
db.session.add(movie)
count += 1
if count % 1000 == 0:
db.session.commit()
print("\r当前进度: {:.2f}%".format(count / 9126 * 100), end="")
db.session.commit()
path = '../ml-latest-small/ratings.csv'
count = 0
fp = open(path, 'r')
fp.readline()
for line in fp.readlines():
ilt = line.split(',')
rating = Rating(user_id=ilt[0], movie_id=ilt[1], rating=ilt[2])
count += 1
if count == 1000:
db.session.commit()
db.session.add(rating)
print("\r当前进度: {:.2f}%".format(count / 100005 * 100), end="")
db.session.commit()
path = '../ml-latest-small/ratings.csv'
count = 0
fp = open(path, 'r')
fp.readline()
for line in fp.readlines():
ilt = line.split(',')
if User.query.filter_by(user_id=ilt[0]).first() is not None:
continue
user = User(user_id=ilt[0], username='******' + str(ilt[0]))
count += 1
if count == 1000:
db.session.commit()
db.session.add(user)
print("\r当前进度: {:.2f}%".format(count / 100005 * 100), end="")
db.session.commit()
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.mkdirs(app.config["UP_DIR"])
os.chmod(app.config["UP_DIR"], "rw")
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 = Movie(
title=data['title'],
#url=url,
#info=data['info'],
#logo=logo,
#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(" added succefully", "ok")
return redirect(url_for("admin.movie_add"))
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")
# 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)
# url,logo为上传视频,图片之后获取到的地址
movie = Movie(title=data["title"],
url=data["url"],
info=data["info"],
logo=logo,
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)
def movie_add():
form = MovieForm()
if form.validate_on_submit():
data = form.data
file_url = secure_filename(form.url.data.filename)
logo_url = 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")
url = change_filename(file_url)
logo = change_filename(logo_url)
form.url.data.save(app.config["UP_DIR"] + url)
form.logo.data.save(app.config["UP_DIR"] + logo)
movie = Movie(
title=data["title"],
url=url,
info=data["info"],
logo=logo,
star=int(data["star"]),
playnum=0,
commentnum=0,
tag_id=int(data["tag_id"]),
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)
def movie_add():
form = MovieForm()
if form.validate_on_submit():
data = form.data
file_url = secure_filename(form.url.data.filename)
file_logo = 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')
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)
move = Movie(title=data['title'],
url=url,
info=data['info'],
logo=logo,
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(move)
db.session.commit()
flash('电影添加成功', 'ok')
return redirect(url_for('admin.movie_add'))
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') # 授予读写权限
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 = Movie(
title=data["title"],
url=url,
info=data["info"],
logo=logo,
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)
def movie_add():
form = MovieForm()
if request.method == 'POST':
data = form.data
if not os.path.exists(app.config['UP_DIR']):
os.makedirs(app.config['UP_DIR'], mode=0o777)
file_url = secure_filename(form.moviefile.data.filename)
file_logo = secure_filename(form.logo.data.filename)
movieurl = change_filename(file_url)
movielogo = change_filename(file_logo)
form.moviefile.data.save(app.config['UP_DIR'] + movieurl)
form.logo.data.save(app.config['UP_DIR'] + movielogo)
movie = Movie(
title=data['title'],
url=movieurl,
info=data['info'],
logo=movielogo,
star=data['star'],
tag_id=data['tag'],
area=data['area'],
release_time=data['release_time'],
length=data['length'],
playnum=0,
commentnum=0,
)
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)
def movie_add():
movie_form = MovieForm()
if movie_form.validate_on_submit():
movie_data = movie_form.data
file_url = secure_filename(
movie_form.url.data.filename) # 使用安全文件名称secure_filename()
print(file_url)
file_log = secure_filename(movie_form.pages.data.filename)
# 判断是否存在保存文件的目录，不存在就创建，并授权读写
if not os.path.exists(app.config['UP_DIR']):
os.makedirs(app.config["UP_DIR"])
os.chmod(app.config["UP_DIR"], "rw")
url = change_file(file_url)
print(url)
page = change_file(file_log)
print(url, page)
# 保存文件到目录
movie_form.url.data.save(app.config["UP_DIR"] + url)
movie_form.pages.data.save(app.config["UP_DIR"] + page)
movie = Movie(title=movie_data['title'],
url=url,
info=movie_data["info"],
logo=page,
star=int(movie_data['star']),
palynum=0,
commentnum=0,
tag_id=int(movie_data['tag_id']),
area=movie_data['area'],
relase_time=movie_data['release_time'],
lenth=movie_data['length'])
db.session.add(movie)
db.session.commit()
flash("添加电影成功", "ok")
return redirect(url_for("admin.movie_add"))
return render_template("hdmin/movie_add.html", form=movie_form)
def movie_add():
movieform = MovieForm()
if movieform.validate_on_submit():
data = movieform.data
file_url = secure_filename(movieform.url.data.filename)
file_logo = secure_filename(movieform.logo.data.filename)
if not os.path.exists(current_app.config['UP_DIR']):
os.mkdir(current_app.config['UP_DIR'])
os.chmod(current_app.config['UP_DIR'], 'rw')
url = change_filename(file_url)
logo = change_filename(file_logo)
movieform.url.data.save(current_app.config['UP_DIR'] + url)
movieform.logo.data.save(current_app.config['UP_DIR'] + logo)
movie = Movie(title=data['title'],
url=url,
info=data['info'],
logo=logo,
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('添加电影成功', 'success')
return redirect(url_for('admin_blueprint.movie_add'))
return render_template('admin/movie_add.html', movieform=movieform)
def movie_add():
movie_form = MovieForm()
if movie_form.validate_on_submit():
data = movie_form.data
file_url = secure_filename(movie_form.url.data.filename)
file_logo = secure_filename(movie_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")
url = change_filename(file_url)
logo = change_filename(file_logo)
# 数据保存
movie_form.url.data.save(app.config["UP_DIR"] + url)
movie_form.logo.data.save(app.config["UP_DIR"] + logo)
movie = Movie(title=data["title"],
url=url,
logo=logo,
star=int(data["star"]),
info=data["info"],
playnums=0,
commentnums=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', movie_form=movie_form)
def movie_add():
form = MovieForm()
if form.validate_on_submit():
data = form.data
movie_count = Movie.query.filter_by(
title=data["title"]).count() #不可以有重复标签
if movie_count == 1:
flash(u"对应的影片已存在！", 'err')
return redirect(url_for('admin.preview_add'))
file_url = secure_filename(form.url.data.filename)
file_logo = secure_filename(form.logo.data.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")
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 = Movie(title=data["title"],
url=url,
info=data["info"],
logo=logo,
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(u"添加电影成功", 'OK')
return redirect(url_for('admin.movie_add'))
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") # 给文件夹赋予读写的权限
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)
# 之后的url和logo就是我们修改之后的地址
movie = Movie(title=data["title"],
url=url,
info=data["info"],
logo=logo,
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)
def movie_add():
# 添加电影页面
form = MovieForm()
if form.validate_on_submit():
data = form.data
res = allowed_file(form.logo.data.filename)
if res == False:
flash(u"图片类型不支持！", "err")
return render_template("admin/movie_add.html", form=form)
file_url = secure_filename(change_filename(form.url.data.filename))
file_logo = secure_filename(change_filename(form.logo.data.filename))
check_upload_dir()
# 保存
form.url.data.save(app.config['UPLOAD_DIR'] + file_url)
form.logo.data.save(app.config['UPLOAD_DIR'] + file_logo)
movie = Movie(title=data["title"],
url=file_url,
info=data["info"],
logo=file_logo,
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)
try:
db.session.commit()
flash(u"添加电影成功！", "ok")
except:
flash(u"添加电影失败！", "err")
return render_template("admin/movie_add.html", form=form)
def setUp(self):
"""Set up method taht will run before every Test
"""
self.new_movie = Movie(
1234, 'Python must be crazy', 'A thrilling new python series',
'https://developers.themoviedb.org/3/getting-started/images/khsjha27hbs',
8.5, 129993)
def setUp(self):
'''
Set up method that will run before every Test
'''
self.new_movie = Movie(1234, "Shazam", "A boy becomes the man",
'https://image.tmdb.org/t/p/w500/khsjha27hbs',
8.5, 129993)
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.url.data.filename)
if not os.path.exists(app.config["UP_DIR"]):
os.makedirs(app.config["UP_DIR"])
os.chmod(app.config["UP_DIR"], "rw")
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 = Movie(
title=data["title"],
url=url,
logo=logo,
playnum=0,
commentnum=0,
author="admin",
tag_id=int(data["tag"]),
)
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)
def setUp(self):
'''
So as to set up the tests correctly before they run
'''
self.new_movie = Movie(234, 'Python Must Be Crazy',
'A thrilling new Python Series', 'khsjha27hbs',
8.5, 129993)
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")
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 = Movie(title=data['title'],
url=url,
info=data['info'],
logo=logo,
star=int(data['star']),
playnum=0,
commentnum=0,
tag_id=int(data['tag_id']),
area=data['area'],
releasetime=data["release_time"],
length=data['length'])
db.session.add(movie)
db.session.commit()
oplog = OpLog(admin_id=session["aid"],
ip=request.remote_addr,
reason="add movie " + movie.title)
db.session.add(oplog)
db.session.commit()
flash("movie add success", "ok")
return redirect(url_for('admin.movie_add'))
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"])
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 = Movie(
title=data["title"],
url=url,
info=data["info"],
logo=logo,
star=int(data["star"]),
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)
def movie_add():
movieform = MovieForm()
if movieform.validate_on_submit():
movieform.tag_id.choices = [(v.id, v.name) for v in Tag.query.all()]
data = movieform.data
file_url = secure_filename(movieform.url.data.filename)
file_logo = secure_filename(movieform.logo.data.filename)
if not os.path.exists(app.config["UP_DIR"]):
os.makedirs(app.config["UP_DIR"])
url = change_filename(file_url)
logo = change_filename(file_logo)
movieform.url.data.save(app.config["UP_DIR"] + url)
movieform.logo.data.save(app.config["UP_DIR"] + logo)
movie = Movie(title=data["title"],
url=url,
info=data["info"],
logo=logo,
star=int(data["star"]),
playnum=0,
commentnum=0,
tag_id=int(data["tag_id"]),
area=data["area"],
release_time=data["release_time"],
lenght=data["length"])
db.session.add(movie)
db.session.commit()
flash("电影添加成功", "ok")
util.InsertOplog("添加电影成功:" + movie.title)
return redirect(url_for("admin.movie_add"))
return render_template("admin/movie_add.html", form=movieform)
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") # 给文件夹授权
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 = Movie(title=data["title"],
url=url,
info=data["info"],
logo=logo,
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")
oplog = Oplog(admin_id=session["admin_id"],
ip=request.remote_addr,
reason="添加电影%s" % data["title"])
db.session.add(oplog)
db.session.commit()
return redirect(url_for('admin.movie_add'))
return render_template("admin/movie_add.html", form=form)
def setUp(self):
'''
set up method that will run for every test
'''
self.new_movie = Movie(1234, "Python Must Be Crazy",
"A Thrilling New Python Series", "khsjha27hbs",
8.5, 129993)
def insertMovieData():
movieDao = MovieDao()
if movieDao.count() == 0:
movie_cols = [
'movie id', 'movie title', 'release date', 'video release date',
'IMDb URL', 'unknown', 'Action', 'Adventure', 'Animation',
'Children\'s', 'Comedy', 'Crime', 'Documentary', 'Drama',
'Fantasy', 'Film-Noir', 'Horror', 'Musical', 'Mystery', 'Romance',
'Sci-Fi', 'Thriller', 'War', 'Western'
]
moviesDataFrame = pd.read_csv('util/dataset/u.item',
sep='|',
names=movie_cols,
encoding='latin-1')
movies = []
for line in moviesDataFrame.itertuples():
genres = getGenreFromArray(line[6:25], movie_cols)
movies.append(
Movie(id=line[1],
title=line[2],
release_date=line[3],
genres=genres))
movieDao.addAll(movies)
else:
print("Movies already loaded!")
def movie_add():
form = MovieForm()
if form.validate_on_submit():
url = form.url.data.filename
logo = form.logo.data.filename
file_save_path = app.config['MOVIE_DIR']
if not os.path.exists(file_save_path):
os.makedirs(file_save_path)
url = change_file(url)
logo = change_file(logo)
form.url.data.save(os.path.join(file_save_path, url))
form.logo.data.save(os.path.join(file_save_path, logo))
movie = Movie(name=form.name.data,
info=form.info.data,
star=form.star.data,
area=form.area.data,
length=form.length.data,
create_time=form.release_time,
url=url,
logo=logo,
tag_id=form.tag_id.data)
db.session.add(movie)
db.session.commit()
flash('添加电影成功')
writer_adminOpLog('删除电影%s成功！' % (movie.name))
return redirect(url_for('admin.movie_add'))
return render_template('admin/movies/add.html', form=form)
def create_movie():
import csv
movies = csv.reader(open('movie.csv'))
for movie in movies:
name = movie[0]
logo = name + movie[1].split('/')[-1]
actors = movie[2]
release_time = movie[3]
tag = movie[5]
area = movie[6]
length = movie[7]
info = movie[-1]
tag = Tag.query.filter_by(name=tag).first()
if not tag:
tag = Tag(name=tag)
db.session.add(tag)
db.session.commit()
print("添加标签%s成功" % (tag))
movie = Movie(name=name,
logo=logo,
tag_id=tag.id,
release_time=release_time,
area=area,
star=randint(1, 5),
length=length,
info=info,
url='20190324_15395101_videos.mp4')
db.session.add(movie)
db.session.commit()
print("添加电影%s成功" % (name))
def add_to_watch():
try:
validate(request.json, schema=new_favourite_schema)
except Exception as e:
return {"Error": "Please Enter Valid Data"}, 412
movie_id = int(request.json["movieID"]) #cast to int since DB has int type
user_id = get_jwt_identity()
#adding movie if it doesn't exist
movie = Movie.query.filter_by(movie_id=movie_id).first()
if (movie == None):
movie = Movie(movie_id=movie_id)
db.session.add(movie)
db.session.commit()
if (UserToWatch.query.filter_by(user_id=user_id,
movie_id=int(movie_id)).first() != None):
return {"Duplicate Entry": 409}, 409
new_to_watch = UserToWatch(user_id=user_id, movie_id=int(movie_id))
db.session.add(new_to_watch)
db.session.commit()
return {"Success": 200}, 200
def setUp(self):
'''
Set up method that will run before every Test
'''
self.new_movie = Movie(1234, 'Python Must Be Crazy',
'A thrilling new Python Series', '/khsjha27hbs',
8.5, 129993)
def movie_add():
form = MovieForm()
if form.validate_on_submit():
file_url = secure_filename(form.url.data.filename)
file_logo = secure_filename(form.logo.data.filename)
url = change_upload_filename(file_url)
logo = change_upload_filename(file_logo)
form.url.data.save(os.path.join(app.config['UP_MOVIE_DIR'], url))
form.logo.data.save(os.path.join(app.config['UP_PIC_DIR'], logo))
movie = Movie(
title=form.title.data,
url=url,
info=form.info.data,
logo=logo,
star=int(form.star.data),
playnum=0,
commentnum=0,
tag_id=int(form.tag_id.data),
area=form.area.data,
length=form.lenth.data,
release_time=form.release_time.data,
)
db.session.add(movie)
db.session.commit()
flash("添加成功", "ok")
return render_template('admin/movie_add.html', form=form)
def get_movie_info(id):
movie_db = create_connection(Config.MOVIE_INFO_DATABASE_PATH)
data = pd.read_sql_query('SELECT * from db where id = ' + str(id), movie_db)
genres = pd.read_sql_query('SELECT genre_name from genres where id = ' + str(id), movie_db)
genres_list = []
for index in range(0, genres.shape[0]):
genres_list.append(genres.at[index, 'genre_name'])
movie = Movie(
original_title=data.at[0,'original_title'],
poster_path=data.at[0,'poster_path'],
release_date=data.at[0,'release_date'],
genres=genres_list,
adult=data.at[0,'adult'],
imdb_id=data.at[0,'imdb_id'],
original_language=data.at[0,'original_language'],
overview=data.at[0,'overview'],
production_companies=data.at[0,'production_companies'],
runtime=data.at[0,'runtime'],
cast = data.at[0, 'cast'].split(',')
)
return movie
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")
# 给该目录授权，让其可读可写
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 = Movie(
title=data['title'],
url=url,
info=data['info'],
logo=logo,
start=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)
