Beispiel #1
0
def test_unique_filename():
    original_filename = 'filename.txt'
    _, original_ext = os.path.splitext(original_filename)
    result = unique_filename(original_filename)
    result_name, result_ext = os.path.splitext(result)

    assert result != original_filename
    assert result_ext == original_ext
    # UUID is a string from 32 hexadecimal digits
    assert len(result_name) == 32
Beispiel #2
0
def test_unique_filename():
    original_filename = 'filename.txt'
    _, original_ext = os.path.splitext(original_filename)
    result = unique_filename(original_filename)
    result_name, result_ext = os.path.splitext(result)

    assert result != original_filename
    assert result_ext == original_ext
    # UUID is a string from 32 hexadecimal digits
    assert len(result_name) == 32
Beispiel #3
0
def save_file(file_obj, where='uploaded'):
    """
    Saves file to where directory on the server
    """
    # todo: check for errors, types or file_obj properties
    folder = os.path.join(config.STATIC_FOLDER, where)
    if not os.path.exists(folder):
        os.makedirs(folder)

    new_filename = unique_filename(file_obj.filename)
    file_path = os.path.join(folder, new_filename)
    with open(file_path, 'wb') as open_file:
        open_file.write(file_obj.file.read())
    return new_filename  # returns new filename
Beispiel #4
0
def save_file(file_obj, where='uploaded'):
    """
    Saves file to where directory on the server
    """
    # todo: check for errors, types or file_obj properties
    folder = os.path.join(config.STATIC_FOLDER, where)
    if not os.path.exists(folder):
        os.makedirs(folder)

    new_filename = unique_filename(file_obj.filename)
    file_path = os.path.join(folder, new_filename)
    with open(file_path, 'wb') as open_file:
        open_file.write(file_obj.file.read())
    return new_filename  # returns new filename
Beispiel #5
0
def find_missing_images():
    print('Fetching images names from database...')
    images_list = get_images_list()
    images_names = map(os.path.basename, images_list)
    found = []
    print('Starting local search...')
    local_files = get_files_under_dir('/', ('.jpg', '.png', '.gif'))
    print('Found %s local images' % len(local_files))
    for local_image in local_files:
        try:
            encoded_name = unique_filename(local_image)
            if encoded_name in images_names:
                found.append((local_image, encoded_name))
        except UnicodeDecodeError as e:
            print('Error with %s, skipping...' % local_image)
    print('Found: %d images of %d' % (len(found), len(images_list)))
    return found
Beispiel #6
0
def find_missing_images():
    print('Fetching images names from database...')
    images_list = get_images_list()
    images_names = map(os.path.basename, images_list)
    found = []
    print('Starting local search...')
    local_files = get_files_under_dir('/', ('.jpg', '.png', '.gif'))
    print('Found %s local images' % len(local_files))
    for local_image in local_files:
        try:
            encoded_name = unique_filename(local_image)
            if encoded_name in images_names:
                found.append((local_image, encoded_name))
        except UnicodeDecodeError as e:
            print('Error with %s, skipping...' % local_image)
    print('Found: %d images of %d' % (len(found), len(images_list)))
    return found
Beispiel #7
0
def up_file():
    """
    Uploads a picture for the article
    """
    up_file = request.files.get('file')
    web_folder = 'img/article/'
    pictures_folder = static_path(web_folder)
    if not os.path.exists(pictures_folder):
        os.makedirs(pictures_folder)

    new_filename = unique_filename(up_file.filename)
    file_path = os.path.join(pictures_folder, new_filename)
    web_path = join_all_path(['/', web_folder, new_filename])
    if os.path.exists(file_path):
        # Just return it, already here
        return web_path

    with open(file_path, 'wb') as open_file:
        open_file.write(up_file.file.read())
    return web_path
Beispiel #8
0
def test_unique_filename_works_for_unicode():
    result = unique_filename(u'should_be_fine.jpg')
    assert isinstance(result, str)
    assert result.endswith('.jpg')