Fixes: #3232. Use of datetime consistent with module_utils/basic.py

This commit is contained in:
Geoff Wright 2016-03-12 10:15:03 +00:00
parent f46ebf2c30
commit e7d3d6c673

View file

@ -98,7 +98,7 @@ import traceback
import SocketServer
from datetime import datetime
import datetime
from threading import Thread, Lock
# import module snippets
@ -257,7 +257,7 @@ class LocalSocketThread(Thread):
# shutdown sooner than expected for new cliets
try:
self.server.last_event_lock.acquire()
self.server.last_event = datetime.now()
self.server.last_event = datetime.datetime.now()
finally:
self.server.last_event_lock.release()
except Exception, e:
@ -293,7 +293,7 @@ class ThreadWithReturnValue(Thread):
class ThreadedTCPServer(SocketServer.ThreadingTCPServer):
key_list = []
last_event = datetime.now()
last_event = datetime.datetime.now()
last_event_lock = Lock()
def __init__(self, server_address, RequestHandlerClass, module, password, timeout, use_ipv6=False):
self.module = module
@ -322,7 +322,7 @@ class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
def send_data(self, data):
try:
self.server.last_event_lock.acquire()
self.server.last_event = datetime.now()
self.server.last_event = datetime.datetime.now()
finally:
self.server.last_event_lock.release()
@ -364,7 +364,7 @@ class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
try:
self.server.last_event_lock.acquire()
self.server.last_event = datetime.now()
self.server.last_event = datetime.datetime.now()
finally:
self.server.last_event_lock.release()
@ -402,15 +402,15 @@ class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
mode = data['mode']
response = {}
last_pong = datetime.now()
last_pong = datetime.datetime.now()
if mode == 'command':
vvvv("received a command request, running it")
twrv = ThreadWithReturnValue(target=self.command, args=(data,))
twrv.start()
response = None
while twrv.is_alive():
if (datetime.now() - last_pong).seconds >= 15:
last_pong = datetime.now()
if (datetime.datetime.now() - last_pong).seconds >= 15:
last_pong = datetime.datetime.now()
vvvv("command still running, sending keepalive packet")
data2 = json.dumps(dict(pong=True))
data2 = self.active_key.Encrypt(data2)
@ -589,7 +589,7 @@ def daemonize(module, password, port, timeout, minutes, use_ipv6, pid_file):
try:
try:
server.last_event_lock.acquire()
td = datetime.now() - server.last_event
td = datetime.datetime.now() - server.last_event
# older python timedelta objects don't have total_seconds(),
# so we use the formula from the docs to calculate it
total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6