common.py

import os
import sys
import time
import unittest


_files = (
    'keys.dat', 'debug.log', 'messages.dat', 'knownnodes.dat',
    '.api_started', 'unittest.lock'
)


def cleanup(home=None, files=_files):
    """Cleanup application files"""
    if not home:
        import state
        home = state.appdata
    for pfile in files:
        try:
            os.remove(os.path.join(home, pfile))
        except OSError:
            pass


def checkup():
    """Checkup files in the src dir"""
    src_dir = os.path.abspath(
        os.path.join(os.path.dirname(__file__), os.pardir))
    for f in _files:
        if os.path.isfile(os.path.join(src_dir, f)):
            return 'Found application file %s in src dir' % f


def skip_python3():
    """Raise unittest.SkipTest() if detected python3"""
    if sys.hexversion >= 0x3000000:
        raise unittest.SkipTest('Module is not ported to python3')


def put_signal_file(path, filename):
    """Creates file, presence of which is a signal about some event."""
    with open(os.path.join(path, filename), 'wb') as outfile:
        outfile.write(b'%i' % time.time())