utils.py

Go to the documentation of this file.

#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-
# @COPYRIGHT_begin
#
# Copyright [2010-2014] Institute of Nuclear Physics PAN, Krakow, Poland
#
# 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.
#
# @COPYRIGHT_end

##
# @package src.ec2
# 
# @copyright Copyright (c) 2012 Institute of Nuclear Physics PAS <http://www.ifj.edu.pl/>
# @author Łukasz Chrząszcz <l.chrzaszcz@gmail.com>
# 
import os
from ec2.settings import BUCKETS_PATH, USER_QUOTA
from ec2.error import InvalidURI


def get_user_used_space(user_name):
    user_path = os.path.join(BUCKETS_PATH, user_name)

    total_size = 0
    for dirpath, dirnames, filenames in os.walk(user_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            total_size += os.path.getsize(fp)

    print 'Used space:', total_size
    return total_size

def get_user_free_space(user_name):
    # config unit is Gb but Content-Length is given in bytes so USER_QUOTA*1024^3
    free_space = USER_QUOTA*1024*1024*1024 - get_user_used_space(user_name)
    return free_space

def check_user_quota(user_name, chunk_size):
    print 'Chunk size:', chunk_size
    free_space = get_user_free_space(user_name)
    print 'Free space:', free_space
    print 'chunk_size type:',type(chunk_size)
    print 'free_space type:',type(free_space)
    if chunk_size <= free_space:
        print 'User has enough space'
        return True
    print 'NO SPACE'
    return False

    # return chunk_size <= free_space

def get_path(user_name, path_info):

    # remove preceding slash to concatenate it with user_name
    if path_info.startswith('/'):
        path_info = path_info[1:]

    # some/bucket/and/object

    path = os.path.join(user_name, path_info)

    # user/some/bucket/and/object

    path = os.path.join(BUCKETS_PATH, path)

    # /usr/lib/cc1/ec2/storage/buckets/user/some/bucket/and/object - unsafe

    path = os.path.abspath(path)

    safe_path = os.path.join(BUCKETS_PATH, user_name)

    if not path.startswith(safe_path):
        raise InvalidURI

    return path

def get_bucket_name(path_info):
    bucket_path = path_info
    slash = bucket_path.find('/')
    if slash != -1:
        bucket_name = bucket_path[:slash+1]
    else:
        bucket_name = bucket_path
    return bucket_name