Exemple #1
0
def consume(method, path, payload=None, cookies=None, params=None):
    _method = method.lower()
    if _method == 'get':
        _method = requests.get
    elif _method == 'post':
        _method = requests.post
    elif _method == 'put':
        _method = requests.put
    elif _method == 'delete' or _method == 'del':
        _method = requests.delete

    _headers = {
        'content-type': 'application/json'
    }

    _payload = json.dumps(payload)
    _cookies = cookies if cookies else storage.memcache('cookie')

    r = _method(path, data=_payload, headers=_headers, cookies=_cookies, params=params)

    if r.status_code == requests.codes.ok:
        return (None, r.json()) if r.content else (None, {})
    else:
        return (r.status_code, r.reason)
Exemple #2
0
def _authenticate():
    res = auth.login(config.username, config.password)
    storage.memcache('cookie', {
        'SID': res['sessionID']
    })
Exemple #3
0
import sys
from os import path
from mdr import importer
from mdr.utils import storage, args


# store module directory path for the future
storage.memcache('module-root', path.dirname(path.abspath(__file__)))

# parse cmd line args
options = args.parse()

# launch
if options.drop:
    importer.drop(options.drops)
elif options.force:
    importer.drop()
    importer.context(options.ctxs)
    importer.extractionSpecification(options.xss)
else:
    importer.context(options.ctxs)
    importer.extractionSpecification(options.xss)