stacktach-winchester/winchester/time_sync.py
Levi Blackstone 8892801205 Add PEP8 check and fix related issues
- Add PEP8 section to tox.ini
- Add hacking to requirements to enforce OpenStack style requirements
- Fix large number of formatting issues flagged by flake8 check
- Add copyright notices to all remaining files
- Fix bug in trigger_manager related to logging calls
- Add .gitignore file

Change-Id: I755ab9c8bcc436836f9006fcd671408cc77214c4
2015-05-04 10:33:25 -05:00

82 lines
2.5 KiB
Python

# Copyright (c) 2014 Dark Secret Software Inc.
# Copyright (c) 2015 Rackspace
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import datetime
import dateutil.parser
import logging
import time
import requests
logger = logging.getLogger(__name__)
class TimeSync(object):
def __init__(self, config={}, publishes=False):
url = config.get('time_sync_endpoint')
self.endpoint = None
if url:
self.endpoint = "%s/time" % url
logger.debug("Time sync endpoint=%s" % self.endpoint)
self.last_update = None
self.last_tyme = self._get_now()
self.publishes = publishes
def _get_now(self):
# Broken out for testing
return datetime.datetime.utcnow()
def _should_update(self, now):
return (not self.last_update or (now - self.last_update).seconds > 20)
def _fetch(self):
while True:
tyme = requests.get(self.endpoint).text
if tyme and tyme != "None":
return tyme
logger.debug("No time published yet. Waiting ...")
time.sleep(1)
def current_time(self):
now = self._get_now()
if not self.endpoint:
return now
if not self.publishes and self._should_update(now):
try:
tyme = self._fetch()
logger.debug("Requested time, got '%s'" % tyme)
self.last_tyme = dateutil.parser.parse(tyme)
except Exception as e:
logger.exception("Could not get time: %s" % e)
self.last_update = now
return self.last_tyme
def publish(self, tyme):
if not self.endpoint:
return
daittyme = dateutil.parser.parse(tyme)
self.last_tyme = daittyme
if self._should_update(daittyme):
self.last_update = daittyme
try:
requests.post(self.endpoint, data=tyme)
logger.debug("Published time: %s" % tyme)
except Exception as e:
logger.exception("Could not publish time: %s" % e)