TypeError:需要类似字节的对象,而不是无服务器和Python3的'str'

时间:2017-11-14 10:09:43

标签: python csv amazon-web-services amazon-s3 serverless-framework

我有一个aws-lambda函数,只要将CSV文件上传到s3存储桶就会触发该函数。我在Python 3.6中使用无服务器框架,问题是我收到此错误消息

  

需要类似字节的对象,而不是'str':TypeError

     

追踪(最近一次呼叫最后一次):

     

文件“/var/task/handler.py”,第33行,在csvfile中

     

fichier = obj ['Body']。read()。split('\ n')

     

TypeError:需要类似字节的对象,而不是'str'

我在网here做了一些研究,问题是,我没有使用open方法导致s3事件读取文件,所以不知道如何修复它

这是我的代码:

import logging
import boto3
from nvd3 import pieChart
import sys
import csv


xdata = []
ydata = []
xdata1 = []
ydata1 = []


logger = logging.getLogger()
logger.setLevel(logging.INFO)

def csvfile(event, context):

    s3 = boto3.client('s3')    
    # retrieve bucket name and file_key from the S3 event
    bucket_name = event['Records'][0]['s3']['bucket']['name']
    file_key = event['Records'][0]['s3']['object']['key']
    logger.info('Reading {} from {}'.format(file_key, bucket_name))
    # get the object
    obj = s3.get_object(Bucket=bucket_name, Key=file_key)
    # get lines inside the csv
    fichier = obj['Body'].read().split('\n')
    #print lines
     for ligne in fichier:
        if len(ligne) > 1:
            logger.info(ligne.decode())
            liste = ligne.split(',')
            print(liste)
            if liste[2] == 'ByCateg':
                xdata.append(liste[4]) 
                ydata.append(liste[1]) 
            elif liste[2] == 'ByTypes':
                xdata1.append(liste[4]) 
                ydata1.append(liste[1]) 

         print ' '.join(xdata) 

print('Function execution Completed')

这是我的serverless.yml代码:

service: aws-python # NOTE: update this with your service name

provider:
  name: aws
  runtime: python3.6
  stage: dev
  region: us-east-1
  iamRoleStatements:
        - Effect: "Allow"
          Action:
              - s3:*
              - "ses:SendEmail"
              - "ses:SendRawEmail"
              - "s3:PutBucketNotification"
          Resource: "*"

    functions:
  csvfile:
    handler: handler.csvfile
    description: send mail whenever a csv file is uploaded on S3 
    events:
      - s3:
          bucket: car2
          event: s3:ObjectCreated:*
          rules:
            - suffix: .csv

1 个答案:

答案 0 :(得分:3)

问题在于

fichier = obj['Body'].read()

返回bytes个对象,而不是字符串。这是因为编码可能需要多个单个字符。现在您在split对象上使用bytes,但无法使用字符串拆分它,您需要使用另一个bytes对象进行拆分。具体来说

fichier = obj['Body'].read().split(b'\n')

应该修正你的错误,但是根据你的预期,分割之前解码会更合适吗?

fichier = obj['Body'].read().decode("utf-8").split('\n')