豆瓣电影影评分析

from peewee import  SqliteDatabase
from models import Comment,Movie,MovieChinese
import pandas  as pd
import jieba
from pyecharts.charts import WordCloud
from pyecharts.globals import ThemeType
import pyecharts.options as opts
from tqdm import tqdm
from cutecharts.charts import Bar


def getMovieInfo(db):
    # 读取'douban_comment_data'数据库
    data = SqliteDatabase(db)
    # 获取'movie'表格中的电影ID,及电影名称 分别添加到movie_id_list , movie_name_list
    movie_id_list = []
    movie_name_list = []
    movie_info = Movie.select()
    # print(movie_info)
    for movie in movie_info:
        movie_id_list.append(movie.id)
        movie_name_list.append(movie.name)

    # 获取movie_chinese表格中的电影ID,及电影名称分别添加到movie_id_list , movie_name_list
    movie_chinese_info = MovieChinese.select()
    for movie_chinese in movie_chinese_info:
        movie_id_list.append(movie_chinese.id)
        movie_name_list.append(movie_chinese.name)
    return movie_name_list,movie_id_list
    # print(movie_name_list)
    # print(movie_id_list)

# 将电影名称和ID做成字典
def getUserChoice(movie_name_list,movie_id_list):
    movie_dict = dict(zip(movie_name_list,movie_id_list))
    # print(movie_dict)

    # 从电影列表中选择一个自己想要看的电影
    your_movie_name = input('输入一个自己想看的电影名称吧>>>')