From 8a4184c0850385280034812414bd7f9c6c3f81b4 Mon Sep 17 00:00:00 2001 From: "Bofu Chen (bafu)" Date: Fri, 5 Apr 2019 14:37:07 +0800 Subject: [PATCH] Save log file at different place to prevent permission error. For root: /tmp/berrynet.log For user: $HOME/.cache/berrynet.log Signed-off-by: Bofu Chen (bafu) --- berrynet/__init__.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/berrynet/__init__.py b/berrynet/__init__.py index 6afbe75..3d6e787 100644 --- a/berrynet/__init__.py +++ b/berrynet/__init__.py @@ -1,4 +1,12 @@ +import os + from logzero import setup_logger -logger = setup_logger(name='berrynet-logger', logfile='/tmp/berrynet.log') +# Save log file at different place to prevent permission error. +if os.geteuid() == 0: # root + LOGGING_FLLEPATH='/tmp/berrynet.log' +else: + LOGGING_FLLEPATH='{}/.cache/berrynet.log'.format(os.getenv('HOME')) + +logger = setup_logger(name='berrynet-logger', logfile=LOGGING_FLLEPATH)