# -*- coding: utf-8 -*-

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import os
from oslo_config.cfg import CONF
import swiftclient.client as swiftclient


def get_swift_session():
    auth_version = CONF.swift.auth_version
    swift_session = swiftclient.Connection(authurl=CONF.swift.auth_uri,
                                           user=CONF.swift.user,
                                           key=CONF.swift.password,
                                           tenant_name=CONF.swift.project,
                                           auth_version=auth_version)
    return swift_session


def upload_raw_content(container, object_name, content):
    swift_session = get_swift_session()
    swift_session.put_object(container, object_name, content)


def download_raw_content(container, object_name):
    swift_session = get_swift_session()
    obj = swift_session.get_object(container, object_name)
    return obj


def upload_to_swift(container, filename, file_id):
    swift_session = get_swift_session()
    with open(os.path.join(filename), 'rb') as upload_file:
        file_id = str(file_id).encode('utf-8')
        swift_session.put_object(container, file_id, upload_file)


def download_from_swift(container, file_uuid):
    swift_session = get_swift_session()
    obj = swift_session.get_object(container, file_uuid)
    with open('/tmp/%s' % file_uuid, 'wb') as download_file:
        download_file.write(obj[1])


def delete_from_swift(container, file_uuid):
    swift_session = get_swift_session()
    swift_session.delete_object(container, file_uuid)