add conf_dir function
This commit is contained in:
parent
1314693ab0
commit
61a48bd8b6
|
@ -1,7 +1,9 @@
|
|||
'''Miscellaneous functions'''
|
||||
import random, string
|
||||
import random, string, sys, os
|
||||
|
||||
from os import environ as env
|
||||
from datetime import datetime
|
||||
from os.path import abspath, dirname, basename, isdir, isfile
|
||||
|
||||
from . import logging
|
||||
|
||||
|
@ -41,6 +43,26 @@ def formatUTC(timestamp=None):
|
|||
return date.strftime('%a, %d %b %Y %H:%M:%S GMT')
|
||||
|
||||
|
||||
def config_dir(modpath=None):
|
||||
if env.get('CONFDIR'):
|
||||
'''set the storage path to the environment variable if it exists'''
|
||||
stor_path = abspath(env['CONFDIR'])
|
||||
|
||||
else:
|
||||
stor_path = f'{os.getcwd()}'
|
||||
|
||||
if modpath and not env.get('CONFDIR'):
|
||||
modname = basename(dirname(modpath))
|
||||
|
||||
if isdir(f'{stor_path}/{modname}'):
|
||||
'''set the storage path to CWD/data if the module or script is in the working dir'''
|
||||
stor_path += '/data'
|
||||
|
||||
if not isdir (stor_path):
|
||||
os.makedirs(stor_path, exist_ok=True)
|
||||
|
||||
return stor_path
|
||||
|
||||
def merp():
|
||||
log = logging.getLogger('merp-heck', {'level': 'merp', 'date': False})
|
||||
log.merp('heck')
|
||||
|
|
Loading…
Reference in a new issue