mirror of
https://github.com/deadc0de6/dotdrop.git
synced 2026-02-04 12:46:44 +00:00
66 lines
1.5 KiB
Python
66 lines
1.5 KiB
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2017, deadc0de6
|
|
helpers for the unittests
|
|
"""
|
|
|
|
import os
|
|
import shutil
|
|
import string
|
|
import random
|
|
import tempfile
|
|
|
|
TMPSUFFIX = '.dotdrop'
|
|
|
|
|
|
def clean(path):
|
|
'''Delete file or folder.'''
|
|
if not os.path.exists(path):
|
|
return
|
|
if os.path.isdir(path):
|
|
shutil.rmtree(path)
|
|
else:
|
|
os.remove(path)
|
|
|
|
|
|
def get_string(length):
|
|
'''Get a random string of length "length".'''
|
|
alpha = string.ascii_uppercase + string.digits
|
|
return ''.join(random.choice(alpha) for _ in range(length))
|
|
|
|
|
|
def get_tempfolder():
|
|
'''Get a temporary folder'''
|
|
return tempfile.mkdtemp(suffix=TMPSUFFIX)
|
|
|
|
|
|
def create_random_file(folder, content=None):
|
|
'''Create a new file in folder with random content.'''
|
|
fname = get_string(8)
|
|
if content is None:
|
|
content = get_string(100)
|
|
path = os.path.join(folder, fname)
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
return path, content
|
|
|
|
|
|
def create_dir(path):
|
|
'''Create a folder'''
|
|
os.mkdir(path)
|
|
return path
|
|
|
|
|
|
def create_fake_config(folder, configname='config.yaml',
|
|
dotpath='dotfiles', backup=True, create=True):
|
|
'''Create a fake config file'''
|
|
path = os.path.join(folder, configname)
|
|
with open(path, 'w') as f:
|
|
f.write('config:\n')
|
|
f.write(' backup: %s\n' % (str(backup)))
|
|
f.write(' create: %s\n' % (str(create)))
|
|
f.write(' dotpath: %s\n' % (dotpath))
|
|
f.write('dotfiles:\n')
|
|
f.write('profiles:\n')
|
|
return path
|