forked from deepjyoti30/ytmdl
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
update: Added logger for better handling.
- Loading branch information
1 parent
311dcf2
commit f50a5d1
Showing
3 changed files
with
164 additions
and
56 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -12,4 +12,6 @@ | |
| 'utility', | ||
| 'gaana', | ||
| 'metadata', | ||
| 'download'] | ||
| 'download', | ||
| 'logger' | ||
| ] | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
|
|
||
| from pathlib import Path | ||
| import datetime | ||
| import os | ||
| from ytmdl.prepend import PREPEND | ||
|
|
||
|
|
||
| class Logger: | ||
| """ | ||
| Custom logger that meets the requirements of using multiple logging setup. | ||
| """ | ||
|
|
||
| def __init__(self, name='', level='INFO'): | ||
| self.name = name | ||
| self._file_format = '' | ||
| self._console_format = '' | ||
| self._log_file = Path('~/.cache/ytmdl/logs/log.cat').expanduser() | ||
| self._check_logfile() | ||
| self._level_number = { | ||
| 'DEBUG': 0, | ||
| 'INFO': 1, | ||
| 'WARNING': 2, | ||
| 'ERROR': 3, | ||
| 'CRITICAL': 4 | ||
| } | ||
| self.level = self._level_number[level] | ||
|
|
||
| def _check_logfile(self): | ||
| """ | ||
| Check if the passed logfile path is present. | ||
| If not present then create it. | ||
| """ | ||
| if not self._log_file.exists(): | ||
| if not self._log_file.parent.exists(): | ||
| os.makedirs(self._log_file.parent) | ||
| f = open(self._log_file, 'w') | ||
| f.close() | ||
|
|
||
| def _write(self, message, LEVEL_NUMBER): | ||
| """ | ||
| Write the logs. | ||
| LEVEL_NUMBER is the levelnumber of the level that is calling the | ||
| _write function. | ||
| """ | ||
| if LEVEL_NUMBER >= self.level: | ||
| self._make_format(message) | ||
| with open(self._log_file, 'a') as f: | ||
| # The file log is to be written to the _log_file file | ||
| f = open(self._log_file, 'a') | ||
| f.write(self._file_format) | ||
| print(self._console_format) | ||
|
|
||
| def _make_format(self, message): | ||
| """ | ||
| Make the format of the string that is to be written. | ||
| """ | ||
| t = datetime.datetime.now() | ||
| DATETIME_FORMAT = '{}-{}-{} {}:{}:{}'.format( | ||
| t.year, | ||
| t.month, | ||
| t.day, | ||
| t.hour, | ||
| t.minute, | ||
| t.second | ||
| ) | ||
| self._console_format = '{}'.format(message) | ||
| self._file_format = '[{}]-[{}]: {}\n'.format(self.name, DATETIME_FORMAT, message) | ||
|
|
||
| def hold(self): | ||
| """ | ||
| Hold the screen by using input() | ||
| """ | ||
| LEVEL_NUMBER = 0 | ||
|
|
||
| if LEVEL_NUMBER >= self.level: | ||
| input("Screen hold! Press any key to continue") | ||
|
|
||
| def debug(self, message): | ||
| """ | ||
| Add the message if the level is debug. | ||
| """ | ||
| LEVEL_NUMBER = 0 | ||
| PREPEND(1) | ||
| self._write(message, LEVEL_NUMBER) | ||
|
|
||
| def info(self, message): | ||
| """ | ||
| Add the message if the level is info or less. | ||
| """ | ||
| LEVEL_NUMBER = 1 | ||
| PREPEND(1) | ||
| self._write(message, LEVEL_NUMBER) | ||
|
|
||
| def warning(self, message): | ||
| """ | ||
| Add the message if the level is warning or less. | ||
| """ | ||
| LEVEL_NUMBER = 2 | ||
| PREPEND(2) | ||
| self._write(message, LEVEL_NUMBER) | ||
|
|
||
| def error(self, message): | ||
| """ | ||
| Add the message if the level is error or less. | ||
| """ | ||
| LEVEL_NUMBER = 3 | ||
| PREPEND(2) | ||
| self._write(message, LEVEL_NUMBER) | ||
|
|
||
| def critical(self, message): | ||
| """ | ||
| Add the message if the level is critical or less. | ||
| """ | ||
| LEVEL_NUMBER = 4 | ||
| PREPEND(2) | ||
| self._write(message, LEVEL_NUMBER) | ||
| exit() |