o i@sdZddlZddlmZddlmZz ddl m Z m Z Wne y-ddl m Z m Z YnwddlZddlZddd Zdd d Zdd d ZdddZdddZdddZdS)uc 电影票房 Created on 2015/12/24 @author: Jimmy Liu @group : waditu @contact: jimmysoa@sina.cn N)cons)dateu)urlopenRequestMbP?c Cst|D]k}t|z'ttjtjdtjdtjt f}t |dd }t |dkr1WdSWnt yH}z t|WYd}~qd}~wwttjrS|dn|}t|d}|jd d gd d }t|d <|SdS)u 获取实时电影票房数据 数据来源：EBOT艺恩票房智库 Parameters ------ retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame BoxOffice 实时票房（万） Irank 排名 MovieName 影片名 boxPer 票房占比 （%） movieDay 上映天数 sumBoxOffice 累计票房（万） time 数据获取时间 httpmbox timeoutNutf-8data2MovieImgmIdaxistime)rangersleeprct MOVIE_BOXP_TYPEDOMAINSBOX_randomrreadlen ExceptionprintjsonloadsPY3decodepd DataFramedropduget_now) retry_countpause_requestlinesejsdfr3M/opt/alphahud/venv/lib/python3.10/site-packages/tushare/internet/boxoffice.pyrealtime_boxoffices(    r5c Cst|D]y}t|z;|durd}n ttt|d}ttj tj dtj dtj |t f}t|dd}t|dkrEWdSWnty\}z t|WYd}~qd}~wwttjrg|d n|}t|d }|jgd dd }|SdS) u1 获取单日电影票房数据 数据来源：EBOT艺恩票房智库 Parameters ------ date:日期，默认为上一日 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame AvgPrice 平均票价 AvpPeoPle 场均人次 BoxOffice 单日票房（万） BoxOffice_Up 环比变化 （%） IRank 排名 MovieDay 上映天数 MovieName 影片名 SumBoxOffice 累计票房（万） WomIndex 口碑指数 Nrrrr r r r rdata1)r BoxOffice1MovieIDDirector IRank_pror)rrrintr)diff_daytodayrr BOXOFFICE_DAYrrrrrrrr r!r"r#r$r%r&r'r( dater+r,r-r.r/r0r1r2r3r3r4 day_boxoffice:s,    rAc Cs|durtddd}n t|dkrttjdS|d7}t|D]d}t|z&t tj tj dtj dtj |f}t|d d }t|d krOWdSWntyf}z t|WYd}~q#d}~wwttjrq|d n|}t|d }|jddgdd}|SdS)u1 获取单月电影票房数据 数据来源：EBOT艺恩票房智库 Parameters ------ date:日期，默认为上一月，格式YYYY-MM retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame Irank 排名 MovieName 电影名称 WomIndex 口碑指数 avgboxoffice 平均票价 avgshowcount 场均人次 box_pro 月度占比 boxoffice 单月票房(万) days 月内天数 releaseTime 上映日期 Nirz-01rr r r r rr6 defaultImage EnMovieIDrr)r) day_last_weekrr!rBOX_INPUT_ERR_MSGrrrrBOXOFFICE_MONTHrrrrrr r"r#r$r%r&r'r(r?r3r3r4month_boxofficehs2     rIcCsj|dur td}t}ttddD]}t||||}|dur*t||g}q| }|j ddS)uz 获取影院单日票房排行数据 数据来源：EBOT艺恩票房智库 Parameters ------ date:日期，默认为上一日 retry_count : int, 默认 3 如遇网络等问题重复执行的次数 pause : int, 默认 0 重复请求数据过程中暂停的秒数，防止请求间隔时间太短出现的问题 return ------- DataFrame Attendance 上座率 AvgPeople 场均人次 CinemaName 影院名称 RowNum 排名 TodayAudienceCount 当日观众人数 TodayBox 当日票房 TodayShowCount 当日场次 price 场均票价（元） Nr T)r() r)rFr&r'r _write_headr _day_cinemaconcatdrop_duplicates reset_index)r@r+r,dataxr2r3r3r4 day_cinemas  rSrc Cstt|D]d}t|z'ttjtjdtjdtj ||f}t |dd }t |dkr5WdSWnt yL}z t|WYd}~qd}~wwttjrW|dn|}t|d} | jdgd d } | SdS) Nrr r r r rr6CinemaIDrr)r_write_consolerrrr BOXOFFICE_CBDrrrrrrr r!r"r#r$r%r&r'r() r@pNor+r,r-r.r/r0r1r2r3r3r4rMs(   rM cCs2ddlm}d|d}d|d}t|||S)Nr)randintr r)randomrYstr)nrYstartendr3r3r4rs   r)rr)Nrr)Nrrr)rX)__doc__pandasr& tushare.stockrr tushare.utilrr)urllib.requestrr ImportErrorurllib2rr"r5rArIrSrMrr3r3r3r4s"     ' . / $