Browse Source

add type info to remaining dataclasses

master
cinnaboot 7 years ago
parent
commit
296990342c
  1. 10
      video_metadata/views/file_listing.py
  2. 35
      video_metadata/views/tmdb.py

10
video_metadata/views/file_listing.py

@ -14,11 +14,11 @@ bp = Blueprint('file_listing', __name__, url_prefix='/file_listing')
@dataclass @dataclass
class path_entry: class path_entry:
base_name = '' base_name: str = ''
rel_path = '' # NOTE: fs path relative to config['MOVIE_DIR'] rel_path: str = '' # NOTE: fs path relative to config['MOVIE_DIR']
file_extension = '' file_extension: str = ''
is_dir = False is_dir: bool = False
in_db = False in_db: bool = False
@bp.route('/folder_contents/') @bp.route('/folder_contents/')

35
video_metadata/views/tmdb.py

@ -7,7 +7,7 @@ import json
import re import re
from sqlite3 import DatabaseError from sqlite3 import DatabaseError
import ssl import ssl
from typing import Dict, List from typing import Any, Dict, List
from urllib.error import URLError, HTTPError from urllib.error import URLError, HTTPError
import urllib.parse import urllib.parse
import urllib.request import urllib.request
@ -48,30 +48,30 @@ class query_data:
full_url: str = '' full_url: str = ''
movie_year: int = 0 movie_year: int = 0
parsed_query_str: str = '' parsed_query_str: str = ''
query_params = {} query_params: str = ''
query_type: query_type = query_type.UNKNOWN query_type: query_type = query_type.UNKNOWN
raw_query_str: str = '' raw_query_str: str = ''
rel_path: str = '' rel_path: str = ''
@dataclass @dataclass
class response_data: class response_data:
errors = [] errors: List[str] = field(default_factory=list)
parsed_response = {} parsed_response: Any = field(default_factory=object)
response_body = '' response_body:str = ''
response_headers = [] response_headers: List[str] = field(default_factory=list)
response_status = 0 response_status:int = 0
result_count = 0 result_count:int = 0
@dataclass @dataclass
class view_result: class view_result:
id = 0 id: int = 0
title = '' title: str = ''
release_date = 0 release_date: str = ''
overview = '' overview: str = ''
genre_ids = [] genre_ids: List[int] = field(default_factory=list)
original_language = '' original_language: str = ''
poster_path = 'no_image.png' poster_path: str = 'no_image.png'
tmdb_link = '' tmdb_link: str = ''
#################### ####################
@ -142,6 +142,9 @@ def query_movie():
storeTMDBQueryResults(query, view_results) storeTMDBQueryResults(query, view_results)
errors.extend(query.errors)
errors.extend(response.errors)
if errors: if errors:
view_results.clear() view_results.clear()
flash(errors) flash(errors)

Loading…
Cancel
Save