ftest_dc_fetchers.py

Go to the documentation of this file.

"""
HCE project, Python bindings, Crawler application.
Fetcher module tests.

@package: dc
@file ftest_dc_fetchers.py
@author bgv <bgv.hce@gmail.com>
@link: http://hierarchical-cluster-engine.com/
@copyright: Copyright &copy; 2015 IOIX Ukraine
@license: http://hierarchical-cluster-engine.com/license/
@since: 1.4.3
"""

import logging
import os
import json
import jsonpickle

from app.Utils import varDump
from dc_crawler.Fetcher import BaseFetcher


# create logger
logger = logging.getLogger('ftest_dc_fetchers')
logger.setLevel(logging.DEBUG)
# create console handler and set level to debug
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
# create formatter
formatter = logging.Formatter('%(asctime)s - %(thread)ld - %(threadName)s - %(module)s - %(funcName)s - %(levelname)s - %(message)s')
# add formatter to ch
ch.setFormatter(formatter)
# add ch to logger
logger.addHandler(ch)


if __name__ == "__main__":
  res = None

  print "CurDir:\n", os.path.dirname(os.path.realpath(__file__))
  loadHeaders = False
  headersDict = {}

  if loadHeaders:
    hdrs = None
    with open("../../ini/crawler-task_headers.txt", 'r') as f:
      hdrs = ''.join(f.readlines())
    for header in hdrs.splitlines():
      if not header:
        continue
      try:
        key, value = header[:header.index(':')].strip(), header[header.index(':') + len(':'):].strip()
      except Exception:
        print "header error:%s", header
        os.abort()
      headersDict[key] = value

  print "headersDict:\n", varDump(headersDict)

  #url = 'http://127.0.0.1/'
  #url = 'http://127.0.0.1/index0.html'
  #url = 'about:blank'
  url = 'https://www.google.co.jp/search?q=&gws_rd=cr###window.IFRAME_KWSRC="http://127.0.0.1/keywords_big.txt";'
  #url = 'https://www.google.com/search?q=test&gws_rd=cr'
  httpTimeout = 60000
  tm = int(httpTimeout) / 1000.0
  if isinstance(httpTimeout, float):
    tm += float('0' + str(httpTimeout).strip()[str(httpTimeout).strip().find('.'):])
  allowRedirects = 1
  proxies = None
  authorization = None
  postData = None
  process_content_types = ["text/html"]
  maxResourceSize = 1024 * 1024
  maxHttpRedirects = 3
  fetchType = BaseFetcher.TYP_NORMAL
  localFilters = None
  urlObjDepth = 0

  #Dynamic fetcher test
  headersDict = {'--disable-web-security':'', '--allow-running-insecure-content':''}
  macroCode = {"name":"tests",
               "sets":[{"name":"set1", "items":['', '', '', ''], "repeat":1, "delay":0}],
               "result_type":2,
               "result_content_type":"text/json",
               "result_fetcher_type":1}

  macroCode['sets'][0]['items'][0] = '1'
  macroCode['sets'][0]['items'][1] = \
  "\
  var s=window.document.createElement('script');\
  s.src='http://127.0.0.1/macro_test4.js';\
  s.type='text/javascript';\
  window.document.head.appendChild(s);\
  return [window.jQuery===undefined, window.MACRO_PREPARE===undefined, window.MACRO_COLLECT===undefined];\
  "
  macroCode['sets'][0]['items'][2] = '50'
  macroCode['sets'][0]['items'][3] = \
  "\
  if(window.MACRO_COLLECT===undefined){\
    return [window.jQuery===undefined, window.MACRO_COLLECT===undefined];\
  }else{\
    return [window.jQuery===undefined, window.MACRO_COLLECT([window.IFRAME_NAME, window.IFRAME_URLS])];\
  }\
  "
  fetchType = BaseFetcher.TYP_DYNAMIC
  #change current dir for webdriver executable run with path ./
  os.chdir("../../bin/")

  try:
    #Test of NORMAL (request lib based) fetcher
    res = BaseFetcher.get_fetcher(fetchType).open(url, timeout=tm, headers=headersDict,
                                                  allow_redirects=allowRedirects, proxies=proxies,
                                                  auth=authorization, data=postData, log=logger,
                                                  allowed_content_types=process_content_types,
                                                  max_resource_size=maxResourceSize,
                                                  max_redirects=maxHttpRedirects,
                                                  filters=localFilters, depth=urlObjDepth, macro=macroCode)

  except Exception, err:
    #logger.debug("Exception:\n%s", varDump(err))
    print "Exception:\n", varDump(err)

  #rd = varDump(res)
  rd = json.dumps(json.loads(jsonpickle.encode(res)), indent=2)
  #logger.debug("Result:\n%s", varDump(res))
  print "Result:\n", rd