Python setup_defaults Examples

Example #1

File: __init__.py Project: wirretheman/svtplay-dl

def main():
    """ Main program """
    parse, options = parser(__version__)

    if options.flexibleq and not options.quality:
        logging.error("flexible-quality requires a quality")

    if len(options.urls) == 0:
        parse.print_help()
        sys.exit(0)
    urls = options.urls
    config = parsertoconfig(setup_defaults(), options)
    if len(urls) < 1:
        parse.error("Incorrect number of arguments")
    setup_log(config.get("silent"), config.get("verbose"))

    if options.cmoreoperatorlist:
        config = parsertoconfig(setup_defaults(), options)
        c = Cmore(config, urls)
        c.operatorlist()
        sys.exit(0)

    try:
        if len(urls) == 1:
            get_media(urls[0], config, __version__)
        else:
            get_multiple_media(urls, config)
    except KeyboardInterrupt:
        print("")
    except (yaml.YAMLError, yaml.MarkedYAMLError) as e:
        logging.error(
            "Your settings file(s) contain invalid YAML syntax! Please fix and restart!, {}"
            .format(str(e)))
        sys.exit(2)

Example #2

File: formatname.py Project: olof/debian-svtplay-dl

 def test_formatname(self):
     for item in self.all_combo:
         config = setup_defaults()
         config.set("filename", item[0])
         service = Service(config, "localhost")
         service.output.update(item[1])
         self.assertEqual(_formatname(service.output, config, "mp4"), item[2])

Example #3

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatname(self):
     for item in self.all_combo:
         config = setup_defaults()
         config.set("filename", item[0])
         service = Service(config, "localhost")
         service.output.update(item[1])
         assert _formatname(service.output, config, "mp4") == item[2]

Example #4

File: test_postprocess.py Project: wirretheman/svtplay-dl

 def test_sublang(self):
     config = setup_defaults()
     config.set("output", os.path.join(os.path.dirname(os.path.realpath(__file__)), "postprocess/textfile-service"))
     service = Service(config, "http://exmaple.com")
     service.output["title"] = "textfile"
     stream = VideoRetriever(config, "http://example.com", 0, output=service.output)
     assert _sublanguage(stream, config, None) == ["swe"]

Example #5

File: __init__.py Project: wirretheman/svtplay-dl

    def __init__(self, config, _url, http=None):
        self._url = _url
        self._urldata = None
        self._error = False
        self.subtitle = None
        self.cookies = {}
        self.auto_name = None
        self.output = {
            "title": None,
            "season": None,
            "episode": None,
            "episodename": None,
            "id": None,
            "service": self.__class__.__name__.lower(),
            "tvshow": None,
            "title_nice": None,
            "showdescription": None,
            "episodedescription": None,
            "showthumbnailurl": None,
            "episodethumbnailurl": None,
            "publishing_datetime": None,
        }
        if not http:
            self.http = HTTP(config)
        else:
            self.http = http

        #  Config
        if config.get("configfile") and os.path.isfile(config.get("configfile")):
            self.config = merge(
                readconfig(setup_defaults(), config.get("configfile"), service=self.__class__.__name__.lower()).get_variable(), config.get_variable()
            )
        else:
            self.config = config
        logging.debug("service: {}".format(self.__class__.__name__.lower()))

Example #6

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameTvshow(self):
     config = setup_defaults()
     service = Service(config, "http://localhost")
     service.output["tvshow"] = True
     service.output["title"] = "kalle"
     service.output["season"] = 2
     service.output["episode"] = 2
     assert formatname(service.output, config) == "kalle.s02e02-service.mp4"

Example #7

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameOutput(self):
     config = setup_defaults()
     config.set("output", "/tmp")
     service = Service(config, "http://localhost")
     if platform.system() == "Windows":
         assert formatname(service.output, config) == "/tmp.mp4"
     else:
         assert formatname(service.output, config) == "/tmp/-service.mp4"

Example #8

File: test_dash.py Project: UWTD/svtplay-dl

def parse(playlist):
    with open(
            os.path.join(os.path.dirname(os.path.realpath(__file__)),
                         "dash-manifests", playlist)) as fd:
        manifest = fd.read()

    return _dashparse(setup_defaults(), manifest, "http://localhost", None,
                      None)

Example #9

File: test_postprocess.py Project: SYSophie/svtplay-dl

 def test_sublang3(self, req):
     config = setup_defaults()
     self.setup_mock(req, "smj")
     config.set("output", os.path.join(os.path.dirname(os.path.realpath(__file__)), "postprocess/textfile-service"))
     config.set("get_all_subtitles", True)
     service = Service(config, "http://exmaple.com")
     service.output["title"] = "textfile"
     stream = VideoRetriever(config, "http://example.com", 0, output=service.output)
     self.assertEqual(_sublanguage(stream, config, ["lulesamiska"]), ["smj"])

Example #10

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameTvshowSubfolderMovie(self):
     config = setup_defaults()
     config.set("subfolder", True)
     service = Service(config, "http://localhost")
     service.output["tvshow"] = False
     service.output["title"] = "kalle"
     service.output["season"] = 2
     service.output["episode"] = 2
     if platform.system() == "Windows":
         assert formatname(service.output, config) == r"movies\kalle.s02e02-service.mp4"
     else:
         assert formatname(service.output, config) == "movies/kalle.s02e02-service.mp4"

Example #11

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameTvshowPath(self):
     config = setup_defaults()
     if platform.system() == "Windows":
         config.set("path", "c:")
     else:
         config.set("path", "/tmp")
     service = Service(config, "http://localhost")
     service.output["title"] = "kalle"
     service.output["season"] = 2
     service.output["episode"] = 2
     if platform.system() == "Windows":
         assert formatname(service.output, config) == r"c:kalle.s02e02-service.mp4"
     else:
         assert formatname(service.output, config) == "/tmp/kalle.s02e02-service.mp4"

Example #12

 def test_sort(self):
     data = [
         DASH(setup_defaults(), "http://example.com", 3000, None),
         HLS(setup_defaults(), "http://example.com", 2000, None),
         HTTP(setup_defaults(), "http://example.com", 3001, None),
     ]
     assert all([
         a[0] == b.bitrate for a, b in zip(
             sort_quality(data),
             [
                 HTTP(setup_defaults(), "http://example.com", 3001, None),
                 DASH(setup_defaults(), "http://example.com", 3000, None),
                 HLS(setup_defaults(), "http://example.com", 2000, None),
             ],
         )
     ])

Example #13

File: __init__.py Project: olof/debian-svtplay-dl

    def __init__(self, config, _url, http=None):
        self._url = _url
        self._urldata = None
        self._error = False
        self.subtitle = None
        self.cookies = {}
        self.auto_name = None
        self.output = {"title": None, "season": None, "episode": None, "episodename": None,
                       "id": None, "service": self.__class__.__name__.lower(),
                       "tvshow": None, "title_nice": None, "showdescription": None,
                       "episodedescription": None, "showthumbnailurl": None,
                       "episodethumbnailurl": None, "publishing_datetime": None}
        if not http:
            self.http = HTTP(config)
        else:
            self.http = http

        #  Config
        if os.path.isfile(config.get("configfile")):
            self.config = merge(readconfig(setup_defaults(), config.get("configfile"),
                                           service=self.__class__.__name__.lower()).get_variable(), config.get_variable())
        else:
            self.config = config
        logging.debug("service: {}".format(self.__class__.__name__.lower()))

Example #14

File: test_text.py Project: wirretheman/svtplay-dl

 def test_exlude_default(self):
     config = setup_defaults()
     assert not exclude(config, "hoppsan")

Example #15

File: service.py Project: olof/debian-svtplay-dl

 def test_service_handler(self):
     config = setup_defaults()
     self.assertIsNone(service_handler(sites, config, "localhost"))

Example #16

File: test_text.py Project: wirretheman/svtplay-dl

 def test_exclude_true(self):
     config = setup_defaults()
     config.set("exclude", "hej")
     assert exclude(config, "hejsanhoppsan")

Example #17

File: test_text.py Project: wirretheman/svtplay-dl

 def test_exclude_false(self):
     config = setup_defaults()
     config.set("exclude", "hej")
     assert not exclude(config, "hoppsan")

Example #18

 def test_vimeo(self):
     config = setup_defaults()
     generic = Generic(config, "http://example.com")
     data = 'src="https://player.vimeo.com/video/359281775" '
     assert isinstance(generic._match(data, sites)[1], Service)

Example #19

File: dash.py Project: olof/debian-svtplay-dl

def parse(playlist):
    with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), "dash-manifests", playlist)) as fd:
        manifest = fd.read()

    return _dashparse(setup_defaults(), manifest, "http://localhost", None, None)

Example #20

 def test_hls(self):
     config = setup_defaults()
     generic = Generic(config, "http://example.com")
     data = 'source src="http://example.com/hls.m3u8" type="application/x-mpegURL"'
     assert isinstance(generic._match(data, sites)[1], Service)

Example #21

 def test_tv4(self):
     config = setup_defaults()
     generic = Generic(config, "http://example.com")
     data = "rc=https://www.tv4play.se/iframe/video/12499319 "
     assert isinstance(generic._match(data, sites)[1], Service)

Example #22

 def test_service_handler(self):
     config = setup_defaults()
     assert isinstance(
         service_handler(sites, config, "https://www.svtplay.se"), Service)

Example #23

 def test_nothing(self):
     config = setup_defaults()
     generic = Generic(config, "http://example.com")
     data = "hejsan"
     assert generic._match(data, sites) == ("http://example.com", None)

Example #24

 def test_service_handler(self):
     config = setup_defaults()
     assert not service_handler(sites, config, "localhost")

Example #25

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameEmpty(self):
     config = setup_defaults()
     service = Service(config, "http://localhost")
     assert formatname(service.output, config) == "-service.mp4"

Example #26

File: service.py Project: olof/debian-svtplay-dl

 def test_service_handler(self):
     config = setup_defaults()
     self.assertIsInstance(service_handler(sites, config, "https://www.svtplay.se"), Service)

Example #27

File: test_formatname.py Project: toran4/svtplay-dl

 def test_formatnameBasedir(self):
     config = setup_defaults()
     service = Service(config, "http://localhost")
     service.output["basedir"] = True
     assert formatname(service.output, config) == "-service.mp4"

Example #28

File: service.py Project: uncoldice/svtplay-dl

 def test_service_handler(self):
     config = setup_defaults()
     self.assertIsNone(service_handler(sites, config, "localhost"))