utils.html 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
  2. "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
  3. <html xmlns="http://www.w3.org/1999/xhtml" lang="en">
  4. <head>
  5. <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  6. <title>proxypool.utils &#8212; proxypool documentation</title>
  7. <link rel="stylesheet" href="../../_static/alabaster.css" type="text/css" />
  8. <link rel="stylesheet" href="../../_static/pygments.css" type="text/css" />
  9. <script type="text/javascript">
  10. var DOCUMENTATION_OPTIONS = {
  11. URL_ROOT: '../../',
  12. VERSION: '',
  13. COLLAPSE_INDEX: false,
  14. FILE_SUFFIX: '.html',
  15. HAS_SOURCE: true
  16. };
  17. </script>
  18. <script type="text/javascript" src="../../_static/jquery.js"></script>
  19. <script type="text/javascript" src="../../_static/underscore.js"></script>
  20. <script type="text/javascript" src="../../_static/doctools.js"></script>
  21. <link rel="index" title="Index" href="../../genindex.html" />
  22. <link rel="search" title="Search" href="../../search.html" />
  23. <link rel="top" title="proxypool documentation" href="../../index.html" />
  24. <link rel="up" title="Module code" href="../index.html" />
  25. <link rel="stylesheet" href="../../_static/custom.css" type="text/css" />
  26. <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
  27. </head>
  28. <body role="document">
  29. <div class="document">
  30. <div class="documentwrapper">
  31. <div class="bodywrapper">
  32. <div class="body" role="main">
  33. <h1>Source code for proxypool.utils</h1><div class="highlight"><pre>
  34. <span></span><span class="kn">import</span> <span class="nn">requests</span>
  35. <span class="kn">import</span> <span class="nn">lxml</span>
  36. <span class="kn">import</span> <span class="nn">asyncio</span>
  37. <span class="kn">import</span> <span class="nn">time</span>
  38. <span class="kn">import</span> <span class="nn">aiohttp</span>
  39. <span class="kn">from</span> <span class="nn">bs4</span> <span class="k">import</span> <span class="n">BeautifulSoup</span>
  40. <span class="n">headers</span> <span class="o">=</span> <span class="p">{</span>
  41. <span class="s1">&#39;User-Agent&#39;</span><span class="p">:</span> <span class="s1">&#39;Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 </span><span class="se">\</span>
  42. <span class="s1"> (KHTML, like Gecko) Chrome/54.0.2840.71 Safari/537.36&#39;</span><span class="p">,</span>
  43. <span class="s1">&#39;Accept-Encoding&#39;</span><span class="p">:</span> <span class="s1">&#39;gzip, deflate, sdch&#39;</span><span class="p">,</span>
  44. <span class="s1">&#39;Accept-Language&#39;</span><span class="p">:</span> <span class="s1">&#39;zh-CN,zh;q=0.8&#39;</span>
  45. <span class="p">}</span>
  46. <div class="viewcode-block" id="get_page"><a class="viewcode-back" href="../../proxypool.html#proxypool.utils.get_page">[docs]</a><span class="k">def</span> <span class="nf">get_page</span><span class="p">(</span><span class="n">url</span><span class="p">):</span>
  47. <span class="n">r</span> <span class="o">=</span> <span class="n">requests</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">url</span><span class="p">,</span> <span class="n">headers</span><span class="o">=</span><span class="n">headers</span><span class="p">)</span>
  48. <span class="k">try</span><span class="p">:</span>
  49. <span class="n">soup</span> <span class="o">=</span> <span class="n">BeautifulSoup</span><span class="p">(</span><span class="n">r</span><span class="o">.</span><span class="n">content</span><span class="o">.</span><span class="n">decode</span><span class="p">(</span><span class="s2">&quot;utf-8&quot;</span><span class="p">),</span> <span class="s1">&#39;lxml&#39;</span><span class="p">)</span>
  50. <span class="k">except</span> <span class="ne">UnicodeDecodeError</span><span class="p">:</span>
  51. <span class="n">soup</span> <span class="o">=</span> <span class="n">BeautifulSoup</span><span class="p">(</span><span class="n">r</span><span class="o">.</span><span class="n">text</span><span class="p">,</span> <span class="s1">&#39;lxml&#39;</span><span class="p">)</span>
  52. <span class="k">return</span> <span class="n">soup</span></div>
  53. <div class="viewcode-block" id="Downloader"><a class="viewcode-back" href="../../proxypool.html#proxypool.utils.Downloader">[docs]</a><span class="k">class</span> <span class="nc">Downloader</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
  54. <span class="sd">&quot;&quot;&quot;</span>
  55. <span class="sd"> 一个异步下载器,可以对代理源异步抓取,但是容易被BAN。</span>
  56. <span class="sd"> &quot;&quot;&quot;</span>
  57. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">urls</span><span class="p">):</span>
  58. <span class="bp">self</span><span class="o">.</span><span class="n">urls</span> <span class="o">=</span> <span class="n">urls</span>
  59. <span class="bp">self</span><span class="o">.</span><span class="n">_htmls</span> <span class="o">=</span> <span class="p">[]</span>
  60. <div class="viewcode-block" id="Downloader.download_single_page"><a class="viewcode-back" href="../../proxypool.html#proxypool.utils.Downloader.download_single_page">[docs]</a> <span class="k">async</span> <span class="k">def</span> <span class="nf">download_single_page</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="p">):</span>
  61. <span class="k">async</span> <span class="k">with</span> <span class="n">aiohttp</span><span class="o">.</span><span class="n">ClientSession</span><span class="p">()</span> <span class="k">as</span> <span class="n">session</span><span class="p">:</span>
  62. <span class="k">async</span> <span class="k">with</span> <span class="n">session</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">url</span><span class="p">)</span> <span class="k">as</span> <span class="n">resp</span><span class="p">:</span>
  63. <span class="bp">self</span><span class="o">.</span><span class="n">_htmls</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="k">await</span> <span class="n">resp</span><span class="o">.</span><span class="n">text</span><span class="p">())</span></div>
  64. <div class="viewcode-block" id="Downloader.download"><a class="viewcode-back" href="../../proxypool.html#proxypool.utils.Downloader.download">[docs]</a> <span class="k">def</span> <span class="nf">download</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  65. <span class="n">loop</span> <span class="o">=</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">get_event_loop</span><span class="p">()</span>
  66. <span class="n">tasks</span> <span class="o">=</span> <span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">download_single_page</span><span class="p">(</span><span class="n">url</span><span class="p">)</span> <span class="k">for</span> <span class="n">url</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">urls</span><span class="p">]</span>
  67. <span class="n">loop</span><span class="o">.</span><span class="n">run_until_complete</span><span class="p">(</span><span class="n">asyncio</span><span class="o">.</span><span class="n">wait</span><span class="p">(</span><span class="n">tasks</span><span class="p">))</span></div>
  68. <span class="nd">@property</span>
  69. <span class="k">def</span> <span class="nf">htmls</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  70. <span class="bp">self</span><span class="o">.</span><span class="n">download</span><span class="p">()</span>
  71. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_htmls</span></div>
  72. </pre></div>
  73. </div>
  74. </div>
  75. </div>
  76. <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
  77. <div class="sphinxsidebarwrapper"><div class="relations">
  78. <h3>Related Topics</h3>
  79. <ul>
  80. <li><a href="../../index.html">Documentation overview</a><ul>
  81. <li><a href="../index.html">Module code</a><ul>
  82. </ul></li>
  83. </ul></li>
  84. </ul>
  85. </div>
  86. <div id="searchbox" style="display: none" role="search">
  87. <h3>Quick search</h3>
  88. <form class="search" action="../../search.html" method="get">
  89. <div><input type="text" name="q" /></div>
  90. <div><input type="submit" value="Go" /></div>
  91. <input type="hidden" name="check_keywords" value="yes" />
  92. <input type="hidden" name="area" value="default" />
  93. </form>
  94. </div>
  95. <script type="text/javascript">$('#searchbox').show(0);</script>
  96. </div>
  97. </div>
  98. <div class="clearer"></div>
  99. </div>
  100. <div class="footer">
  101. &copy;2016, Author.
  102. |
  103. Powered by <a href="http://sphinx-doc.org/">Sphinx 1.4.8</a>
  104. &amp; <a href="https://github.com/bitprophet/alabaster">Alabaster 0.7.9</a>
  105. </div>
  106. </body>
  107. </html>