test_image_clients.py 4.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. '''
  2. Function:
  3. Implementation of Dedicated ImageClient
  4. Author:
  5. Zhenchao Jin
  6. WeChat Official Account (微信公众号):
  7. Charles的皮卡丘
  8. '''
  9. from imagedl.modules.sources import (
  10. BingImageClient, I360ImageClient, YahooImageClient, BaiduImageClient, SogouImageClient, GoogleImageClient, YandexImageClient, PixabayImageClient,
  11. DuckduckgoImageClient, UnsplashImageClient, GelbooruImageClient, SafebooruImageClient, DanbooruImageClient, PexelsImageClient, DimTownImageClient,
  12. HuabanImageClient, FoodiesfeedImageClient, EverypixelImageClient, FreeNatureStockImageClient, WeiboImageClient
  13. )
  14. # bing tests
  15. client = BingImageClient()
  16. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  17. client.download(image_infos, num_threadings=1)
  18. # 360 tests
  19. client = I360ImageClient()
  20. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  21. client.download(image_infos, num_threadings=1)
  22. # baidu tests
  23. client = BaiduImageClient()
  24. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  25. client.download(image_infos, num_threadings=1)
  26. # sogou tests
  27. client = SogouImageClient()
  28. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  29. client.download(image_infos, num_threadings=1)
  30. # google tests
  31. client = GoogleImageClient()
  32. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  33. client.download(image_infos, num_threadings=1)
  34. # yandex tests
  35. client = YandexImageClient()
  36. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  37. client.download(image_infos, num_threadings=1)
  38. # pixabay tests
  39. client = PixabayImageClient()
  40. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  41. client.download(image_infos, num_threadings=1)
  42. # duckduckgo tests
  43. client = DuckduckgoImageClient()
  44. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  45. client.download(image_infos, num_threadings=1)
  46. # yahoo tests
  47. client = YahooImageClient()
  48. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  49. client.download(image_infos, num_threadings=1)
  50. # unsplash tests
  51. client = UnsplashImageClient()
  52. image_infos = client.search('Cute Dogs', search_limits=10, num_threadings=1)
  53. client.download(image_infos, num_threadings=1)
  54. # gelbooru tests
  55. client = GelbooruImageClient()
  56. image_infos = client.search('pikachu', search_limits=10, num_threadings=1)
  57. client.download(image_infos, num_threadings=1)
  58. # safebooru tests
  59. client = SafebooruImageClient()
  60. image_infos = client.search('pikachu', search_limits=10, num_threadings=1)
  61. client.download(image_infos, num_threadings=1)
  62. # danbooru tests
  63. client = DanbooruImageClient()
  64. image_infos = client.search('pikachu', search_limits=10, num_threadings=1)
  65. client.download(image_infos, num_threadings=1)
  66. # pexels tests
  67. client = PexelsImageClient()
  68. image_infos = client.search('animals', search_limits=10, num_threadings=1)
  69. client.download(image_infos, num_threadings=1)
  70. # dimtown tests
  71. client = DimTownImageClient()
  72. image_infos = client.search('JK', search_limits=10, num_threadings=1)
  73. client.download(image_infos, num_threadings=1)
  74. # huaban tests
  75. client = HuabanImageClient()
  76. image_infos = client.search('JK', search_limits=10, num_threadings=1)
  77. client.download(image_infos, num_threadings=1)
  78. # foodiesfeed tests
  79. client = FoodiesfeedImageClient()
  80. image_infos = client.search('pizza', search_limits=10, num_threadings=1)
  81. client.download(image_infos, num_threadings=1)
  82. # everypixel tests (cookies required)
  83. client = EverypixelImageClient(default_search_cookies='xxxx')
  84. image_infos = client.search('animals', search_limits=10, num_threadings=1)
  85. client.download(image_infos, num_threadings=1)
  86. # freenaturestock tests
  87. client = FreeNatureStockImageClient()
  88. image_infos = client.search('mountains', search_limits=10, num_threadings=1)
  89. client.download(image_infos, num_threadings=1)
  90. # weibo tests (cookies required)
  91. client = WeiboImageClient(default_search_cookies='xxxx')
  92. image_infos = client.search('animals', search_limits=10, num_threadings=1)
  93. client.download(image_infos, num_threadings=1)