From 8849fb2b2b8dc824bd3f17ce1026a04856477a94 Mon Sep 17 00:00:00 2001
From: Jordan Wiens <jordan@psifertex.com>
Date: Sun, 4 Mar 2018 18:08:04 -0500
Subject: working division, prints, and metaclasses, but imports broken, still
 needs work

---
 python/update.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

(limited to 'python/update.py')

diff --git a/python/update.py b/python/update.py
index 1eb8ea61..1fda149b 100644
--- a/python/update.py
+++ b/python/update.py
@@ -21,14 +21,17 @@
 import traceback
 import ctypes
 
-# Binary Ninja components
-import _binaryninjacore as core
-from enums import UpdateResult
-import startup
-import log
+# Binary Ninja components -- additional imports belong in the appropriate class
+from binaryninja import _binaryninjacore as core
+from binaryninja.enums import UpdateResult
+
+
+#2-3 compatibility
+from six import with_metaclass
 
 
 class _UpdateChannelMetaClass(type):
+	from binaryninja import startup
 	@property
 	def list(self):
 		startup._init_plugins()
@@ -95,6 +98,7 @@ class _UpdateChannelMetaClass(type):
 
 
 class UpdateProgressCallback(object):
+	from binaryninja import log
 	def __init__(self, func):
 		self.cb = ctypes.CFUNCTYPE(ctypes.c_bool, ctypes.c_void_p, ctypes.c_ulonglong, ctypes.c_ulonglong)(self.callback)
 		self.func = func
@@ -108,8 +112,7 @@ class UpdateProgressCallback(object):
 			log.log_error(traceback.format_exc())
 
 
-class UpdateChannel(object):
-	__metaclass__ = _UpdateChannelMetaClass
+class UpdateChannel(with_metaclass(_UpdateChannelMetaClass, object)):
 
 	def __init__(self, name, desc, ver):
 		self.name = name
-- 
cgit v1.3.1