44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
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())
|