Advertisement
Guest User

Untitled

a guest
Dec 2nd, 2018
119
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 10.42 KB | None | 0 0
  1. root@EFCinema:/opt/Varken/Varken# cat helpers.py
  2. import os
  3. import time
  4. import tarfile
  5. import geoip2.database
  6. from typing import NamedTuple
  7. from os.path import abspath, join
  8. from urllib.request import urlretrieve
  9.  
  10.  
  11. class TVShow(NamedTuple):
  12. seriesId: int = None
  13. episodeFileId: int = None
  14. seasonNumber: int = None
  15. episodeNumber: int = None
  16. title: str = None
  17. airDate: str = None
  18. airDateUtc: str = None
  19. overview: str = None
  20. episodeFile: dict = None
  21. hasFile: bool = None
  22. monitored: bool = None
  23. unverifiedSceneNumbering: bool = None
  24. absoluteEpisodeNumber: int = None
  25. series: dict = None
  26. id: int = None
  27.  
  28.  
  29. class Movie(NamedTuple):
  30. title: str = None
  31. alternativeTitles: list = None
  32. secondaryYearSourceId: int = None
  33. sortTitle: str = None
  34. sizeOnDisk: int = None
  35. status: str = None
  36. overview: str = None
  37. inCinemas: str = None
  38. images: list = None
  39. downloaded: bool = None
  40. year: int = None
  41. secondaryYear: str = None
  42. hasFile: bool = None
  43. youTubeTrailerId: str = None
  44. studio: str = None
  45. path: str = None
  46. profileId: int = None
  47. pathState: str = None
  48. monitored: bool = None
  49. minimumAvailability: str = None
  50. isAvailable: bool = None
  51. folderName: str = None
  52. runtime: int = None
  53. lastInfoSync: str = None
  54. cleanTitle: str = None
  55. imdbId: str = None
  56. tmdbId: int = None
  57. titleSlug: str = None
  58. genres: list = None
  59. tags: list = None
  60. added: str = None
  61. ratings: dict = None
  62. movieFile: dict = None
  63. qualityProfileId: int = None
  64. physicalRelease: str = None
  65. physicalReleaseNote: str = None
  66. website: str = None
  67. id: int = None
  68.  
  69. class Queue(NamedTuple):
  70. movie: dict = None
  71. series: dict = None
  72. episode: dict = None
  73. quality: dict = None
  74. size: float = None
  75. title: str = None
  76. sizeleft: float = None
  77. timeleft: str = None
  78. estimatedCompletionTime: str = None
  79. status: str = None
  80. trackedDownloadStatus: str = None
  81. statusMessages: list = None
  82. downloadId: str = None
  83. protocol: str = None
  84. id: int = None
  85.  
  86.  
  87. class SonarrServer(NamedTuple):
  88. id: int = None
  89. url: str = None
  90. api_key: str = None
  91. verify_ssl: bool = False
  92. missing_days: int = 0
  93. missing_days_run_seconds: int = 30
  94. future_days: int = 0
  95. future_days_run_seconds: int = 30
  96. queue: bool = False
  97. queue_run_seconds: int = 30
  98.  
  99. class RadarrServer(NamedTuple):
  100. id: int = None
  101. url: str = None
  102. api_key: str = None
  103. verify_ssl: bool = False
  104. queue: bool = False
  105. queue_run_seconds: int = 30
  106. get_missing: bool = False
  107. get_missing_run_seconds: int = 30
  108.  
  109.  
  110. class OmbiServer(NamedTuple):
  111. id: int = None
  112. url: str = None
  113. api_key: str = None
  114. verify_ssl: bool = False
  115. request_type_counts: bool = False
  116. request_type_run_seconds: int = 30
  117. request_total_counts: bool = False
  118. request_total_run_seconds: int = 30
  119.  
  120.  
  121. class TautulliServer(NamedTuple):
  122. id: int = None
  123. url: str = None
  124. fallback_ip: str = None
  125. api_key: str = None
  126. verify_ssl: bool = None
  127. get_activity: bool = False
  128. get_activity_run_seconds: int = 30
  129. get_sessions: bool = False
  130. get_sessions_run_seconds: int = 30
  131.  
  132.  
  133. class InfluxServer(NamedTuple):
  134. url: str = 'localhost'
  135. port: int = 8086
  136. username: str = 'root'
  137. password: str = 'root'
  138.  
  139.  
  140. class OmbiRequestCounts(NamedTuple):
  141. pending: int = 0
  142. approved: int = 0
  143. available: int = 0
  144.  
  145.  
  146. class TautulliStream(NamedTuple):
  147. rating: str = None
  148. transcode_width: str = None
  149. labels: list = None
  150. stream_bitrate: str = None
  151. bandwidth: str = None
  152. optimized_version: int = None
  153. video_language: str = None
  154. parent_rating_key: str = None
  155. rating_key: str = None
  156. platform_version: str = None
  157. transcode_hw_decoding: int = None
  158. thumb: str = None
  159. title: str = None
  160. video_codec_level: str = None
  161. tagline: str = None
  162. last_viewed_at: str = None
  163. audio_sample_rate: str = None
  164. user_rating: str = None
  165. platform: str = None
  166. collections: list = None
  167. location: str = None
  168. transcode_container: str = None
  169. audio_channel_layout: str = None
  170. local: str = None
  171. stream_subtitle_format: str = None
  172. stream_video_ref_frames: str = None
  173. transcode_hw_encode_title: str = None
  174. stream_container_decision: str = None
  175. audience_rating: str = None
  176. full_title: str = None
  177. ip_address: str = None
  178. subtitles: int = None
  179. stream_subtitle_language: str = None
  180. channel_stream: int = None
  181. video_bitrate: str = None
  182. is_allow_sync: int = None
  183. stream_video_bitrate: str = None
  184. summary: str = None
  185. stream_audio_decision: str = None
  186. aspect_ratio: str = None
  187. audio_bitrate_mode: str = None
  188. transcode_hw_decode_title: str = None
  189. stream_audio_channel_layout: str = None
  190. deleted_user: int = None
  191. library_name: str = None
  192. art: str = None
  193. stream_video_resolution: str = None
  194. video_profile: str = None
  195. sort_title: str = None
  196. stream_video_codec_level: str = None
  197. stream_video_height: str = None
  198. year: str = None
  199. stream_duration: str = None
  200. stream_audio_channels: str = None
  201. video_language_code: str = None
  202. transcode_key: str = None
  203. transcode_throttled: int = None
  204. container: str = None
  205. stream_audio_bitrate: str = None
  206. user: str = None
  207. selected: int = None
  208. product_version: str = None
  209. subtitle_location: str = None
  210. transcode_hw_requested: int = None
  211. video_height: str = None
  212. state: str = None
  213. is_restricted: int = None
  214. email: str = None
  215. stream_container: str = None
  216. transcode_speed: str = None
  217. video_bit_depth: str = None
  218. stream_audio_sample_rate: str = None
  219. grandparent_title: str = None
  220. studio: str = None
  221. transcode_decision: str = None
  222. video_width: str = None
  223. bitrate: str = None
  224. machine_id: str = None
  225. originally_available_at: str = None
  226. video_frame_rate: str = None
  227. synced_version_profile: str = None
  228. friendly_name: str = None
  229. audio_profile: str = None
  230. optimized_version_title: str = None
  231. platform_name: str = None
  232. stream_video_language: str = None
  233. keep_history: int = None
  234. stream_audio_codec: str = None
  235. stream_video_codec: str = None
  236. grandparent_thumb: str = None
  237. synced_version: int = None
  238. transcode_hw_decode: str = None
  239. user_thumb: str = None
  240. stream_video_width: str = None
  241. height: str = None
  242. stream_subtitle_decision: str = None
  243. audio_codec: str = None
  244. parent_title: str = None
  245. guid: str = None
  246. audio_language_code: str = None
  247. transcode_video_codec: str = None
  248. transcode_audio_codec: str = None
  249. stream_video_decision: str = None
  250. user_id: int = None
  251. transcode_height: str = None
  252. transcode_hw_full_pipeline: int = None
  253. throttled: str = None
  254. quality_profile: str = None
  255. width: str = None
  256. live: int = None
  257. stream_subtitle_forced: int = None
  258. media_type: str = None
  259. video_resolution: str = None
  260. stream_subtitle_location: str = None
  261. do_notify: int = None
  262. video_ref_frames: str = None
  263. stream_subtitle_language_code: str = None
  264. audio_channels: str = None
  265. stream_audio_language_code: str = None
  266. optimized_version_profile: str = None
  267. relay: int = None
  268. duration: str = None
  269. rating_image: str = None
  270. is_home_user: int = None
  271. is_admin: int = None
  272. ip_address_public: str = None
  273. allow_guest: int = None
  274. transcode_audio_channels: str = None
  275. stream_audio_channel_layout_: str = None
  276. media_index: str = None
  277. stream_video_framerate: str = None
  278. transcode_hw_encode: str = None
  279. grandparent_rating_key: str = None
  280. original_title: str = None
  281. added_at: str = None
  282. banner: str = None
  283. bif_thumb: str = None
  284. parent_media_index: str = None
  285. live_uuid: str = None
  286. audio_language: str = None
  287. stream_audio_bitrate_mode: str = None
  288. username: str = None
  289. subtitle_decision: str = None
  290. children_count: str = None
  291. updated_at: str = None
  292. player: str = None
  293. subtitle_format: str = None
  294. file: str = None
  295. file_size: str = None
  296. session_key: str = None
  297. id: str = None
  298. subtitle_container: str = None
  299. genres: list = None
  300. stream_video_language_code: str = None
  301. indexes: int = None
  302. video_decision: str = None
  303. stream_audio_language: str = None
  304. writers: list = None
  305. actors: list = None
  306. progress_percent: str = None
  307. audio_decision: str = None
  308. subtitle_forced: int = None
  309. profile: str = None
  310. product: str = None
  311. view_offset: str = None
  312. type: str = None
  313. audience_rating_image: str = None
  314. audio_bitrate: str = None
  315. section_id: str = None
  316. stream_subtitle_codec: str = None
  317. subtitle_codec: str = None
  318. video_codec: str = None
  319. device: str = None
  320. stream_video_bit_depth: str = None
  321. video_framerate: str = None
  322. transcode_hw_encoding: int = None
  323. transcode_protocol: str = None
  324. shared_libraries: list = None
  325. stream_aspect_ratio: str = None
  326. content_rating: str = None
  327. session_id: str = None
  328. directors: list = None
  329. parent_thumb: str = None
  330. subtitle_language_code: str = None
  331. transcode_progress: int = None
  332. subtitle_language: str = None
  333. stream_subtitle_container: str = None
  334.  
  335.  
  336. def geoip_download():
  337. tar_dbfile = abspath(join('.', 'data', 'GeoLite2-City.tar.gz'))
  338. url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-City.tar.gz'
  339. urlretrieve(url, tar_dbfile)
  340. tar = tarfile.open(tar_dbfile, "r:gz")
  341. for files in tar.getmembers():
  342. if 'GeoLite2-City.mmdb' in files.name:
  343. files.name = os.path.basename(files.name)
  344. tar.extract(files, abspath(join('.', 'data')))
  345. os.remove(tar_dbfile)
  346.  
  347. def geo_lookup(ipaddress):
  348.  
  349. dbfile = abspath(join('.', 'data', 'GeoLite2-City.mmdb'))
  350. now = time.time()
  351.  
  352. try:
  353. dbinfo = os.stat(dbfile)
  354. db_age = now - dbinfo.st_ctime
  355. if db_age > (35 * 86400):
  356. os.remove(dbfile)
  357. geoip_download()
  358. except FileNotFoundError:
  359. geoip_download()
  360.  
  361. reader = geoip2.database.Reader(dbfile)
  362.  
  363. return reader.city(ipaddress)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement