From fb410e168e29ad7698d96d24412e31f1b68fbdcf Mon Sep 17 00:00:00 2001
From: pran1990 <praneshpg@gmail.com>
Date: Mon, 19 May 2014 17:25:53 -0700
Subject: [PATCH] Use six for python 2/3 compatibility

Replace usage of dict.keys() with six.iterkeys(dict)

Change-Id: Ia7c99ea202fd0800c43a079b5b3fd094e7e66adb
---
 entropy/engine.py | 3 ++-
 entropy/utils.py  | 5 +++--
 requirements.txt  | 1 +
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/entropy/engine.py b/entropy/engine.py
index 96575df..8ef5430 100644
--- a/entropy/engine.py
+++ b/entropy/engine.py
@@ -25,6 +25,7 @@ import croniter
 from kombu import Exchange
 from kombu import Queue
 import pause
+import six
 
 from entropy import utils
 import imp
@@ -110,7 +111,7 @@ class Engine(object):
 
             new_additions = []
 
-            for key in schedules.keys():
+            for key in six.iterkeys(schedules):
                 sched = schedules[key]
                 now = datetime.datetime.now()
                 cron = croniter.croniter(sched, now)
diff --git a/entropy/utils.py b/entropy/utils.py
index 859e7b7..75c2e6c 100644
--- a/entropy/utils.py
+++ b/entropy/utils.py
@@ -19,6 +19,7 @@ import os
 import sys
 import time
 
+import six
 from watchdog.events import FileSystemEventHandler
 from watchdog.observers import Observer
 import yaml
@@ -97,7 +98,7 @@ class WatchdogHandler(FileSystemEventHandler):
         self.event_fn = event_fn
 
     def on_modified(self, event):
-        if event.src_path in self.event_fn.keys():
+        if event.src_path in six.iterkeys(self.event_fn):
             self.event_fn[event.src_path]()
         else:
             LOG.error('no associated function for %s', event.src_path)
@@ -113,7 +114,7 @@ def watch_dir_for_change(dir_to_watch, event_fn):
 
 def check_duplicate(name, cfg_file):
     scripts = load_yaml(cfg_file)
-    return scripts and name in scripts.keys()
+    return scripts and name in scripts
 
 
 def reset_logger(log):
diff --git a/requirements.txt b/requirements.txt
index 11e8047..60d61a5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,4 @@ pause==0.1.2
 stevedore>0.10
 watchdog>=0.7.1
 futures>=2.1.6
+six>=1.6.1