diff --git a/src/you_get/downloader/__init__.py b/src/you_get/downloader/__init__.py index a053ef43..ffb534e40 100644 --- a/src/you_get/downloader/__init__.py +++ b/src/you_get/downloader/__init__.py @@ -1,6 +1,5 @@ #!/usr/bin/env python -from .__main__ import * from .acfun import * from .alive import * from .baidu import * @@ -38,3 +37,5 @@ from .xiami import * from .yinyuetai import * from .youku import * from .youtube import * + +from .__main__ import * diff --git a/tests/test.py b/tests/test.py index 2a25779b..641878ef 100644 --- a/tests/test.py +++ b/tests/test.py @@ -4,7 +4,7 @@ import unittest from you_get import * -from you_get.__main__ import url_to_module +from you_get.downloader.__main__ import url_to_module def test_urls(urls): for url in urls: