Update helpers.now function

This commit is contained in:
JonnyWong16 2020-08-02 13:48:10 -07:00
parent 9baecb0a41
commit f582f781f3
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
3 changed files with 13 additions and 12 deletions

View file

@ -17,7 +17,6 @@ from __future__ import unicode_literals
from future.builtins import object from future.builtins import object
from future.builtins import str from future.builtins import str
import arrow
import os import os
import re import re
import shutil import shutil
@ -28,8 +27,10 @@ from configobj import ConfigObj, ParseError
import plexpy import plexpy
if plexpy.PYTHON2: if plexpy.PYTHON2:
import helpers
import logger import logger
else: else:
from plexpy import helpers
from plexpy import logger from plexpy import logger
@ -268,9 +269,9 @@ def make_backup(cleanup=False, scheduler=False):
""" Makes a backup of config file, removes all but the last 5 backups """ """ Makes a backup of config file, removes all but the last 5 backups """
if scheduler: if scheduler:
backup_file = 'config.backup-%s.sched.ini' % arrow.now().format('YYYYMMDDHHmmss') backup_file = 'config.backup-{}.sched.ini'.format(helpers.now(no_sep=True))
else: else:
backup_file = 'config.backup-%s.ini' % arrow.now().format('YYYYMMDDHHmmss') backup_file = 'config.backup-{}.ini'.format(helpers.now(no_sep=True))
backup_folder = plexpy.CONFIG.BACKUP_DIR backup_folder = plexpy.CONFIG.BACKUP_DIR
backup_file_fp = os.path.join(backup_folder, backup_file) backup_file_fp = os.path.join(backup_folder, backup_file)

View file

@ -17,7 +17,6 @@ from __future__ import unicode_literals
from future.builtins import str from future.builtins import str
from future.builtins import object from future.builtins import object
import arrow
import os import os
import sqlite3 import sqlite3
import shutil import shutil
@ -26,11 +25,11 @@ import time
import plexpy import plexpy
if plexpy.PYTHON2: if plexpy.PYTHON2:
import helpers
import logger import logger
from helpers import cast_to_int, chunk
else: else:
from plexpy import helpers
from plexpy import logger from plexpy import logger
from plexpy.helpers import cast_to_int, chunk
FILENAME = "tautulli.db" FILENAME = "tautulli.db"
@ -219,7 +218,7 @@ def delete_recently_added():
def delete_rows_from_table(table, row_ids): def delete_rows_from_table(table, row_ids):
if row_ids and isinstance(row_ids, str): if row_ids and isinstance(row_ids, str):
row_ids = list(map(cast_to_int, row_ids.split(','))) row_ids = list(map(helpers.cast_to_int, row_ids.split(',')))
if row_ids: if row_ids:
logger.info("Tautulli Database :: Deleting row ids %s from %s database table", row_ids, table) logger.info("Tautulli Database :: Deleting row ids %s from %s database table", row_ids, table)
@ -230,7 +229,7 @@ def delete_rows_from_table(table, row_ids):
monitor_db = MonitorDatabase() monitor_db = MonitorDatabase()
try: try:
for row_ids_group in chunk(row_ids, sqlite_max_variable_number): for row_ids_group in helpers.chunk(row_ids, sqlite_max_variable_number):
query = "DELETE FROM " + table + " WHERE id IN (%s) " % ','.join(['?'] * len(row_ids_group)) query = "DELETE FROM " + table + " WHERE id IN (%s) " % ','.join(['?'] * len(row_ids_group))
monitor_db.action(query, row_ids_group) monitor_db.action(query, row_ids_group)
except Exception as e: except Exception as e:
@ -293,9 +292,9 @@ def make_backup(cleanup=False, scheduler=False):
plexpy.NOTIFY_QUEUE.put({'notify_action': 'on_plexpydbcorrupt'}) plexpy.NOTIFY_QUEUE.put({'notify_action': 'on_plexpydbcorrupt'})
if scheduler: if scheduler:
backup_file = 'tautulli.backup-{}{}.sched.db'.format(arrow.now().format('YYYYMMDDHHmmss'), corrupt) backup_file = 'tautulli.backup-{}{}.sched.db'.format(helpers.now(no_sep=True), corrupt)
else: else:
backup_file = 'tautulli.backup-{}{}.db'.format(arrow.now().format('YYYYMMDDHHmmss'), corrupt) backup_file = 'tautulli.backup-{}{}.db'.format(helpers.now(no_sep=True), corrupt)
backup_folder = plexpy.CONFIG.BACKUP_DIR backup_folder = plexpy.CONFIG.BACKUP_DIR
backup_file_fp = os.path.join(backup_folder, backup_file) backup_file_fp = os.path.join(backup_folder, backup_file)

View file

@ -213,9 +213,10 @@ def today():
return yyyymmdd return yyyymmdd
def now(): def now(no_sep=False):
now = datetime.datetime.now() now = datetime.datetime.now()
if no_sep:
return now.strftime("%Y%m%d%H%M%S")
return now.strftime("%Y-%m-%d %H:%M:%S") return now.strftime("%Y-%m-%d %H:%M:%S")