ZendeskPDFMaker.py 文件源码

def post_pdfs_to_s3(self):
    conn = boto.s3.connect_to_region('us-east-1',
                                     aws_access_key_id=S3_ACCESS_KEY_FOR_MANUAL,
                                     aws_secret_access_key=S3_SECRET_KEY_FOR_MANUAL,
                                     calling_format=OrdinaryCallingFormat())
    bucket_name = S3_BUCKET_FOR_MANUAL
    bucket_dir = S3_DIRECTORY_FOR_MANUAL
    bucket = conn.get_bucket(bucket_name, validate=False)
    source_dir = os.path.join(ZENDESK_UTIL_DIR, 'gen/pdf/')
    print "posting pdfs from %s" % source_dir
    section_dict = {}
    for fn in os.listdir(source_dir):
      with open(source_dir + fn, 'r') as pdf_file:
        chunks = fn.split('-')
        category = chunks[0]
        filename = '-'.join(chunks[1:len(chunks)])
        if not category in section_dict:
          section_dict[category] = ''
        section_dict[category] += '<tr><td style="padding-right:10px;padding-bottom:5px"><a href=http://{}/{}/{}/{}>{}</a></td><td>http://{}/{}/{}/{}</td></tr>'.format(bucket_name, bucket_dir, category, filename, filename, bucket_name, bucket_dir, category, filename)
        k = Key(bucket)
        k.key = '/' + bucket_dir + '/' + category + '/' + filename
        print("POSTING PDF to S3: " + k.key)
        k.set_contents_from_file(pdf_file,cb=self.percent_cb, num_cb=1)
    self.post_inventory_html(section_dict, bucket, bucket_name)