Sfoglia il codice sorgente

producer 생성 버그 수정, log formatter 수정

heewon 3 anni fa
parent
commit
c58b9e7e85
1 ha cambiato i file con 7 aggiunte e 8 eliminazioni
  1. 7 8
      flask_app/app.py

+ 7 - 8
flask_app/app.py

@@ -6,29 +6,28 @@ import boto3
 import logging
 import logging.handlers
 import os
+import time
 
 server = Flask(__name__)
 api = Api(server)
-
 session = boto3.Session( 
-        aws_access_key_id='AKIAUSTWEOGGEJICFGIP',
-        aws_secret_access_key='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',
+        aws_access_key_id=os.environ['ID'],
+        aws_secret_access_key=os.environ['KEY'],
         region_name='ap-northeast-2')
-
+producer = KinesisProducer(stream_name=os.environ['STREAM'],boto3_session=session)
 logger = logging.getLogger(__name__)
-formatter = logging.Formatter('%(message)s')
+os.environ['TZ'] = 'Asia/Seoul'
+time.tzset()
+formatter = logging.Formatter('%(message)s') if os.environ['FORMATTER_VER']==1 else logging.Formatter('[%(asctime)s] %(message)s', "%Y-%m-%d %H:%M:%S")
 fileHandler = logging.FileHandler('./logs/log')
 fileHandler.setFormatter(formatter)
 logger.addHandler(fileHandler)
 logger.setLevel(level=logging.INFO)
 class LogWallpad(Resource):
     def post(self):
-        producer = KinesisProducer(stream_name=os.environ['STREAM'],boto3_session=session)
         request.json['site']=os.environ['SITE']
         logger.info(json.dumps(request.json))
         producer.put(json.dumps(request.json)+'\n')
         return request.json, 201
 
 api.add_resource(LogWallpad, '/wallpad/log')
-
-