recommend-emr-dataprocess/rec_scripts/rov/train/src/tools/update2oss.py

#!/usr/bin/env python
# coding=utf-8

import argparse
import os

import oss2
from oss2.credentials import EnvironmentVariableCredentialsProvider

os.environ['OSS_ACCESS_KEY_ID'] = 'LTAIP6x1l3DXfSxm'
os.environ['OSS_ACCESS_KEY_SECRET'] = 'KbTaM9ars4OX3PMS6Xm7rtxGr1FLon'
endpoint = 'https://oss-cn-hangzhou.aliyuncs.com'
bucket_name = 'art-recommend'
region = 'cn-hangzhou'
oss_sub_path = 'zhangbo'


def get_bucket():
    auth = oss2.ProviderAuthV4(EnvironmentVariableCredentialsProvider())
    return oss2.Bucket(auth, endpoint, bucket_name, region=region)


def get_filename(file_path):
    return file_path[file_path.rfind('/') + 1:]


def update(local_file):
    bucket = get_bucket()
    filename = get_filename(local_file)
    oss_dest_file = '%s/%s' % (oss_sub_path, filename)
    bucket.put_object_from_file(oss_dest_file, local_file)
    print('%s -> %s' % (local_file, oss_dest_file))


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--local', required=True, type=str, help='local file')
    args = parser.parse_args()
    print('\n\n')
    print(args)
    print('\n\n')

    if os.path.exists(args.local):
        update(args.local)