Как получить доступ к информации из файла config. json в файл python? - PullRequest
0 голосов
/ 13 января 2020

Проблема в том, что я не могу получить доступ к информации из файла конфигурации. json к моему python файлу. Я предоставил данные JSON и код python ниже. Я попробовал все в модуле запроса, но Я могу получить доступ к ответу без файла конфигурации, но мне нужно с файлом конфигурации. Ниже приведен файл json

{
    "api_data": {
        "request_url": "https://newapi.zivame.com/api/v1/catalog/list",
        "post_data" : {"category_ids" : "948",
                        "limit" : "10000"},
        "my_headers":{"Content-Type": "application/json"}
    },
    "redshift":{
        "host":"XXX.XXXX.XXX",
        "user":"XXXX",
        "password":"XXXXXXXX",
        "port": 8080,
        "db":"XXXX"
    },
    "s3":{
        "access_key":"XXXXXXXXX",
        "secret_key":"XXXXXXXXXX",
        "region":"XX-XXXXX-1",
        "path":"XXXXXXXXXXXX/XXX",
        "table":"XXXXXX",
        "bucket":"XXXX",
        "file": "XXXXXX",
        "copy_column": "XXX",
        "local_path": "XXXXX"
    },
    "csv_file": {
        "promo_zivame": ""
    }
}

, и это программа

#!/usr/bin/python                                                                                                                                                                 
import json                                                                                                                                                                       
import psycopg2                                                                                                                                                                   
import requests                                                                                                                                                                   
import os

BASE_PATH = os.path.dirname(os.path.realpath(__file__))

with open(BASE_PATH+'/config.json') as json_data_file:                                                                                                                            
    data = json.load(json_data_file)

#api_config = data['api_data']                                                                                                                                                    
#redshift = data['redshift']                                                                                                                                                      
s3_config = data['s3']

#x = print(api_config.get('request_url'))

class ApiResponse:                                                                                                                                                                
    #api response                                                                                                                                                                 
    def api_data(self, api_config):                                                                                                                                               
        print("starting api_data")                                                                                                                                                
        try:                                                                                                                                                                      
            self.ApiResponse = requests.post(api_config['request_url'], api_config['post_data'], api_config['my_headers'])                                                        
            data_1 = self.ApiResponse                                                                                                                                             
            #data = json.dump(self.ApiResponse)                                                                                                                                   
            print("API Result Response")                                                                                                                                          
            print(())                                                                                                                                                             
            print(self.ApiResponse)                                                                                                                                               
            return (self.ApiResponse)

        except Exception:                                                                                                                                                         
            print("response not found")                                                                                                                                           
            return False

    def redshift_connect(self, redshift):                                                                                                                                         
        try:                                                                                                                                                                      
            # Amazon Redshift connect string                                                                                                                                      
            self.con = psycopg2.connect(                                                                                                                                          
                host=redshift['host'],                                                                                                                                            
                user=redshift['user'],                                                                                                                                            
                port=redshift['port'],                                                                                                                                            
                password=redshift['password'],                                                                                                                                    
                dbname=redshift['db'])                                                                                                                                            
            print(self.con)                                                                                                                                                       
            return self.con                                                                                                                                                       
        except Exception:                                                                                                                                                         
            print("Error in Redshift connection")                                                                                                                                 
            return False                                                                                                                                                          
def main():                                                                                                                                                                       
    c1 = ApiResponse()                                                                                                                                                                   
    api_config = data['api_data']                                                                                                                                                 
    redshift = data['redshift']                                                                                                                                                   
    c1.api_data(api_config)                                                                                                                                                       
    c1.api_data(data)                                                                                                                                                             
    c1.redshift_connect(redshift)

if __name__=='__main__':                                                                                                                                                          
    main()

Ответы [ 2 ]

0 голосов
/ 14 января 2020

Заимствование кода из { ссылка }

import requests

api_config = {
    "request_url": "https://newapi.zivame.com/api/v1/catalog/list",
    "post_data" : {"category_ids" : "948", "limit" : "10000"},
    "my_headers":{"Content-Type": "application/json"}
    }
local_filename = 'the_response.json'
with requests.post(api_config['request_url'], json=api_config['post_data'], headers=api_config['my_headers'], stream=True) as r:
    r.raise_for_status()
    with open(local_filename, 'wb') as f:
        for chunk in r.iter_content(chunk_size=8192): 
            if chunk: # filter out keep-alive new chunks
                f.write(chunk)

сохраняет ответ в файл ('the_response.json'), который затем можно передать. Обратите внимание, что stream=True передано requests.post

0 голосов
/ 13 января 2020

Третий аргумент requests.post() равен json. Чтобы предоставить headers, вам нужно явно использовать имя аргумента, как предложено @JustinEzequiel. См. Запросы do c здесь: 2. python -requests.org / ru / v1.1.0 / user / quickstart / # custom-headers

requests.post(api_config['request_url'], json=api_config['post_data'], headers=api_config['my_headers'])
...