コード例 #1
0
ファイル: test_http_client.py プロジェクト: devova/pipes
def test_http_response():
    request = Request()
    request.headers = {
        'Authorization': 'token',
        'header1': 'value1',
        'header2': 'value2'
    }
    request.method = 'GET'
    request.url = 'http://url.com'
    request.body = None

    response = HttpResponse()
    response.status_code = 400
    response.request = request

    assert response.to_curl(request) == (
        "curl -X 'GET' -v -H 'Authorization: <...>' "
        "-H 'header1: value1' -H 'header2: value2' "
        "'http://url.com'")

    assert sorted(
        response.hide_and_serialize_sensitive_headers(
            request.headers).split('\n')) == [
                '', 'Authorization: <...>', 'header1: value1',
                'header2: value2'
            ]

    with pytest.raises(ExtendedHTTPError) as exc_info:
        # ensure error is raised
        response.raise_for_status()
    assert exc_info.value.response == response
    assert exc_info.value.request == request
コード例 #2
0
ファイル: db_auto_backup.py プロジェクト: wemecan/School
 def _send_request(method, url, headers, body):
     """发送请求"""
     s = Session()
     prepped = Request(method, url).prepare()
     prepped.headers = headers
     prepped.body = body
     r = s.send(prepped)
     return r.status_code, r.text
コード例 #3
0
ファイル: upyun.py プロジェクト: ping203/School
def main():
    file = None
    with open(r'E:\MyDesktop\default_avatar_female_50.gif', 'rb') as f:
        file = f.read()
    fileMd5 = hashlib.md5(file).hexdigest()
    date = httpdate_rfc1123()
    uri = '/image-upyun-9527/3.webp'
    method = 'PUT'
    headers = {
        'Authorization': sign(key, secret, method, uri, date),
        'Date': date,
        'Content-Length': str(len(file)),
        'x-gmkerl-thumb': '/format/webp/quality/85/progressive/true/strip/true'
    }
    print(headers)
    from requests import Session, Request
    s = Session()
    prepped = Request(method, 'http://v0.api.upyun.com' + uri).prepare()
    prepped.headers = headers
    prepped.body = file
    r = s.send(prepped)
    print(r.status_code, r.text)
コード例 #4
0
def lambda_handler(event, context):
    pp.pprint(event)

    aws_sns_topic_arn = event['sns']
    url = event['url']
    method = event.get('method', 'GET')
    headers = event.get('headers')
    body = event.get('body')

    session = Session()
    request = Request(method, url).prepare()
    if headers is not None:
        request.headers = headers
    if body is not None:
        request.body = body
    response = session.send(request)

    if response.status_code == 200:
        content = response.text
        pp.pprint(content)
        publish_command_to_sns(aws_sns_topic_arn, content)
    else:
        pp.pprint(response)
コード例 #5
0
ファイル: ipaiyun.py プロジェクト: wemecan/School
def main():
    file = None
    with open(
            r'C:\Users\Steven Fang\Documents\Tencent Files\1075004549\Image\Group\{SS]6F@`F[F6WY~[%{F`C_N.gif',
            'rb') as f:
        file = f.read()
    fileMd5 = hashlib.md5(file).hexdigest()
    date = httpdate_rfc1123()
    uri = '/image-upyun-9527/3.webp'
    method = 'PUT'
    headers = {
        'Authorization': sign(key, secret, method, uri, date),
        'Date': date,
        'Content-Length': str(len(file)),
        'x-gmkerl-thumb': '/format/webp/quality/85/progressive/true/strip/true'
    }
    print(headers)
    from requests import Session, Request
    s = Session()
    prepped = Request(method, 'http://v0.api.upyun.com' + uri).prepare()
    prepped.headers = headers
    prepped.body = file
    r = s.send(prepped)
    print(r.status_code, r.text)
コード例 #6
0
    def test_signature_generation_for_iam(self):
        """
        Test the dictionary returned by AWSProfileAuth which is used
        in `auth` input to request sent to AWS API Gateway.
        This dict should contain signature values used by AWS API
        Gateway Resource Policy to determine access to API from HTTP
        request. This test just confirms the override functionality
        we needed to create to support AWS_PROFILE. By default,
        aws_requests_auth doesn't support this directly.
        """
        # Mock request object, needed for signature generation
        r = Request('POST',
                    "https://123456780.execute-api.us-west-2.amazonaws.com",
                    json={'key': 'value'})
        r.prepare()
        r.body = "test-text"

        sig_generation_without_profile = kubectl_hyperkube.AWSProfileAuth(
            aws_host="https://123456780.execute-api.us-west-2.amazonaws.com",
            aws_region='us-west-2',
            aws_service='execute-api',
            profile=None)

        sig_without_profile_result = (
            sig_generation_without_profile.get_aws_request_headers(
                r,
                aws_access_key="fffasdfdfadadsfdkaf101010",
                aws_secret_access_key="akdsfjasdfwiqer;s123k122",
                aws_token="testingtokenaklzaner"))
        assert "Credential=fffasdfdfadadsfdkaf101010" in (
            sig_without_profile_result['Authorization'])
        assert "us-west-2/execute-api/aws4_request" in (
            sig_without_profile_result['Authorization'])
        assert ("SignedHeaders=host;x-amz-date;x-amz-security-token"
                in (sig_without_profile_result['Authorization']))
        assert (
            "b18939c1891ba923a4d46ccf146612094490300d22e436bf5a5d6d46d99e4225"
            in (  # noqa
                sig_without_profile_result['x-amz-content-sha256']))
        assert sig_without_profile_result['X-Amz-Security-Token'] == (
            "testingtokenaklzaner")

        # requires ~/.aws/config to have a `[default]` profile
        sig_generation_with_profile = kubectl_hyperkube.AWSProfileAuth(
            aws_host="https://123456780.execute-api.us-west-2.amazonaws.com",
            aws_region='us-west-2',
            aws_service='execute-api',
            profile="default")

        sig_with_profile_result = (
            sig_generation_with_profile.get_aws_request_headers(
                r,
                aws_access_key="fffasdfdfadadsfdkaf102381",
                aws_secret_access_key="akdsfjasdfwiqer;sfkjk213",
                aws_token="testingtokenakldsfkja;"))
        assert "Credential=fffasdfdfadadsfdkaf102381" in (
            sig_with_profile_result['Authorization'])
        assert "us-west-2/execute-api/aws4_request" in (
            sig_with_profile_result['Authorization'])
        assert ("SignedHeaders=host;x-amz-date;x-amz-security-token"
                in (sig_with_profile_result['Authorization']))
        assert (
            "b18939c1891ba923a4d46ccf146612094490300d22e436bf5a5d6d46d99e4225"
            in (  # noqa
                sig_with_profile_result['x-amz-content-sha256']))
        assert sig_with_profile_result['X-Amz-Security-Token'] == (
            "testingtokenakldsfkja;")
コード例 #7
0
ファイル: RWSHelper.py プロジェクト: s546360316/cms
def main():
    parser = argparse.ArgumentParser(prog='cmsRWSHelper')
    parser.add_argument(
        '-v', '--verbose', action='store_true',
        help="tell on stderr what's happening")
    # FIXME It would be nice to use '--rankings' with action='store'
    # and nargs='+' but it doesn't seem to work with subparsers...
    parser.add_argument(
        '-r', '--ranking', dest='rankings', action='append', default=None,
        choices=list(xrange(len(config.rankings_address))), metavar='shard',
        type=int, help="select which RWS to connect to (omit for 'all')")
    subparsers = parser.add_subparsers(
        title='available actions', metavar='action',
        help='what to ask the RWS to do with the entity')

    # Create the parser for the "get" command
    parser_get = subparsers.add_parser('get', help="retrieve the entity")
    parser_get.set_defaults(action='get')

    # Create the parser for the "create" command
    parser_create = subparsers.add_parser('create', help="create the entity")
    parser_create.set_defaults(action='create')
    parser_create.add_argument(
        'file', type=argparse.FileType('rb'),
        help="file holding the entity body to send ('-' for stdin)")

    # Create the parser for the "update" command
    parser_update = subparsers.add_parser('update', help='update the entity')
    parser_update.set_defaults(action='update')
    parser_update.add_argument(
        'file', type=argparse.FileType('rb'),
        help="file holding the entity body to send ('-' for stdin)")

    # Create the parser for the "delete" command
    parser_delete = subparsers.add_parser('delete', help='delete the entity')
    parser_delete.set_defaults(action='delete')

    # Create the group for entity-related arguments
    group = parser.add_argument_group(
        title='entity reference')
    group.add_argument(
        'entity_type', action='store', choices=ENTITY_TYPES, metavar='type',
        help="type of the entity (e.g. contest, user, task, etc.)")
    group.add_argument(
        'entity_id', action='store', type=six.text_type, metavar='id',
        help='ID of the entity (usually a short codename)')

    # Parse the given arguments
    args = parser.parse_args()

    args.entity_id = quote(args.entity_id)

    if args.verbose:
        verb = args.action[:4] + 'ting'
        logger.info("%s entity '%ss/%s'" % (verb.capitalize(),
                                            args.entity_type, args.entity_id))

    if args.rankings is not None:
        shards = args.rankings
    else:
        shards = list(xrange(len(config.rankings_address)))

    s = Session()
    error = False

    for shard in shards:
        url, username, password = get_parameters(
            shard, args.entity_type, args.entity_id)

        if args.verbose:
            logger.info(
                "Preparing %s request to %s (username: %s; password: %s)" %
                (ACTION_METHODS[args.action], url, username, password))

        req = Request(ACTION_METHODS[args.action],
                      url, auth=(username, password)).prepare()

        if hasattr(args, 'file'):
            if args.verbose:
                logger.info("Reading file contents to use as message body")
            req.body = args.file.read()

        if args.verbose:
            logger.info("Sending request")

        try:
            res = s.send(req, verify=config.https_certfile)
        except RequestException as e:
            logger.error("Failed")
            logger.info(repr(e))
            error = True
            continue

        if args.verbose:
            logger.info("Response received")

        if res.status_code != (201 if args.action == "create" else 200):
            logger.error("Unexpected status code: %d" % res.status_code)
            error = True
            continue

        if args.action == "get":
            print(res.content)

    if error:
        sys.exit(1)