def test_check_file_extension():
    for ext in ['txt', '.txt', 'TXT', '.TXT', 'tXt', '.tXt', 'TxT', '.TxT']:
        assert check_file_extension('file.txt', ext) == ('file', '.txt')

    assert check_file_extension('file.json', 'txt', 'yaml') == (None, None)
    assert check_file_extension('file.json', 'txt', 'yaml',
                                'json') == ('file', '.json')
def test_check_file_extension_dir(tmpdir):
    assert check_file_extension(str(tmpdir.join('file.txt')),
                                'txt') == ('file', '.txt')
    assert check_file_extension(str(tmpdir.join('file.txt')),
                                'json') == (None, None)
    assert check_file_extension(str(tmpdir.join('file.txt'))) == (None, None)
    assert check_file_extension(str(tmpdir.join('out')), 'txt') == (None, None)

    assert tmpdir.remove() is None
def test_check_file_extension():
    for ext in ['txt', '.txt', 'TXT', '.TXT', 'tXt', '.tXt', 'TxT', '.TxT']:
        assert check_file_extension('file.txt', ext) == ('file', '.txt')

    assert check_file_extension(
        'file.json', 'txt', 'yaml'
    ) == (None, None)
    assert check_file_extension(
        'file.json', 'txt', 'yaml', 'json'
    ) == ('file', '.json')
def test_check_file_extension_dir(tmpdir):
    assert check_file_extension(
        str(tmpdir.join('file.txt')), 'txt'
    ) == ('file', '.txt')
    assert check_file_extension(
        str(tmpdir.join('file.txt')), 'json'
    ) == (None, None)
    assert check_file_extension(
        str(tmpdir.join('file.txt'))
    ) == (None, None)
    assert check_file_extension(
        str(tmpdir.join('out')), 'txt'
    ) == (None, None)

    assert tmpdir.remove() is None
示例#5
0
def _sidecar_path(ff, sc):
    '''
    Check passed sidecars for valid paths, format (*json* or *yaml*) and for
    valid filenames (no double dots).

    :param ff: running :class:`ffflash.main.FFFlash` instance
    :param sc: sidecar as passed by user
    :return: Tuple of either (``False``, ``None``, ``None``) on error or:

        * normalized and full path to ``sc``
        * unvalidated key-names into api-file
        * ``True`` if ``sc`` is a *yaml* file, ``False`` if it's *json*

    '''
    ff.log('handling sidecar {}'.format(sc))

    sidepath = check_file_location(sc)
    if not sidepath:
        return ff.log('sidecar {} is either a folder, or parent folder does '
                      'not exist yet skipping'.format(sc),
                      level=False), None, None

    name, ext = check_file_extension(sidepath, 'yaml', 'json')
    if not all([name, ext]):
        return ff.log('sidecar {} is neither json nor yaml'.format(sc),
                      level=False), None, None

    fields = name.split('.')
    if not all([f for f in fields]):
        return ff.log('sidecar {} {} name is invalid '
                      '(check for double dots)'.format(sc, name),
                      level=False), None, None

    ff.log('sidecar path {} is valid. got {}'.format(sidepath, fields))
    return sidepath, fields, (True if ext == '.yaml' else False)
def test_check_file_extension_no_ext():
    assert check_file_extension('file') == (None, None)
    assert check_file_extension('file', '') == (None, None)

    assert check_file_extension('file.txt') == (None, None)
    assert check_file_extension('file.txt', '') == (None, None)
    assert check_file_extension('file.txt', '.') == (None, None)
    assert check_file_extension('file.txt', '..') == (None, None)

    assert check_file_extension('file.txt', 'file') == (None, None)
    assert check_file_extension('file', 'file') == (None, None)
def test_check_file_extension_no_ext():
    assert check_file_extension('file') == (None, None)
    assert check_file_extension('file', '') == (None, None)

    assert check_file_extension('file.txt') == (None, None)
    assert check_file_extension('file.txt', '') == (None, None)
    assert check_file_extension('file.txt', '.') == (None, None)
    assert check_file_extension('file.txt', '..') == (None, None)

    assert check_file_extension('file.txt', 'file') == (None, None)
    assert check_file_extension('file', 'file') == (None, None)
示例#8
0
def _rankfile_load(ff):
    '''
    Load either existing ``rankfile`` from disk, or create empty stub
    if one does not exist yet. Path and extension (*json*)
    get validated.

    :param ff: running :class:`ffflash.main.FFFlash` instance
    :return: Tuple of either (``False``, ``None``) on error or:

        * validated path to the ``rankfile``
        * ``rankfile`` content
    '''
    if not ff.access_for('rankfile'):
        return (False, None)
    ff.log('handling rankfile {}'.format(ff.args.rankfile))

    rankfile = check_file_location(ff.args.rankfile, must_exist=False)
    if not rankfile:
        return ff.log(
            'wrong path for rankfile {}'.format(ff.args.rankfile),
            level=False
        ), None

    if not all(check_file_extension(rankfile, 'json')):
        return ff.log(
            'rankfile {} is no json'.format(rankfile),
            level=False
        ), None

    ranks = load_file(rankfile, fallback={
        'updated_at': 'never', 'nodes': []
    }, as_yaml=False)

    if not ranks or not isinstance(ranks, dict):
        return ff.log(
            'could not load rankfile {}'.format(rankfile),
            level=False
        ), None

    if not all([(a in ranks) for a in ['nodes', 'updated_at']]):
        return ff.log(
            'this is no rankfile {}'.format(rankfile),
            level=False
        ), None

    lranks = len(ranks.get('nodes', 0))
    ff.log((
        'creating new rankfile {}'.format(rankfile)
        if lranks == 0 else
        'loaded {} nodes'.format(lranks)
    ))
    return rankfile, ranks
示例#9
0
def _sidecar_path(ff, sc):
    '''
    Check passed sidecars for valid paths, format (*json* or *yaml*) and for
    valid filenames (no double dots).

    :param ff: running :class:`ffflash.main.FFFlash` instance
    :param sc: sidecar as passed by user
    :return: Tuple of either (``False``, ``None``, ``None``) on error or:

        * normalized and full path to ``sc``
        * unvalidated key-names into api-file
        * ``True`` if ``sc`` is a *yaml* file, ``False`` if it's *json*

    '''
    ff.log('handling sidecar {}'.format(sc))

    sidepath = check_file_location(sc)
    if not sidepath:
        return ff.log(
            'sidecar {} is either a folder, or parent folder does '
            'not exist yet skipping'.format(sc),
            level=False
        ), None, None

    name, ext = check_file_extension(sidepath, 'yaml', 'json')
    if not all([name, ext]):
        return ff.log(
            'sidecar {} is neither json nor yaml'.format(sc),
            level=False
        ), None, None

    fields = name.split('.')
    if not all([f for f in fields]):
        return ff.log(
            'sidecar {} {} name is invalid '
            '(check for double dots)'.format(sc, name),
            level=False
        ), None, None

    ff.log('sidecar path {} is valid. got {}'.format(sidepath, fields))
    return sidepath, fields, (True if ext == '.yaml' else False)