rjames86 · May 23, 2015 18:59 · May 23, 2015
diff --git a/vnc_favorites.py b/vnc_favorites.py
@@ -0,0 +1,129 @@
+from xml.etree.ElementTree import Element, SubElement, tostring
+from os import path
+
+from Foundation import (
+    CFPreferencesAppSynchronize,
+    CFPreferencesCopyAppValue,
+    CFPreferencesCopyKeyList,
+    CFPreferencesSetValue,
+    kCFPreferencesAnyUser,
+    kCFPreferencesCurrentHost
+)
+
+HOME = path.expanduser('~')
+
+PLIST_PATH = HOME + "/Library/Preferences/com.apple.sidebarlists.plist"
+
+
+class toXML():
+
+    def __init__(self):
+        self.toxml = Element('items')
+
+    def __repr__(self):
+        return tostring(self.toxml)
+
+    def add_item(self, title, subtitle="", arg="", valid="YES", autocomplete="", icon="icon.png"):
+        item = SubElement(self.toxml, 'item', {
+            'uid': str(len(self.toxml)),
+            'arg': arg,
+            'valid': valid,
+            'autocomplete': autocomplete
+            })
+
+        _title = SubElement(item, 'title')
+        _title.text = title
+
+        _subtitle = SubElement(item, 'subtitle')
+        _subtitle.text = subtitle
+
+        _icon = SubElement(item, 'icon')
+        _icon.text = icon
+
+    def display(self):
+        print self
+
+
+class Preferences(object):
+    """Class which directly reads/writes Apple CF preferences."""
+
+    def __init__(self, bundle_id, user=kCFPreferencesAnyUser):
+        """Init.
+
+        Args:
+            bundle_id: str, like 'ManagedInstalls'
+        """
+        if bundle_id.endswith('.plist'):
+            bundle_id = bundle_id[:-6]
+        self.bundle_id = bundle_id
+        self.user = user
+
+    def __iter__(self):
+        keys = CFPreferencesCopyKeyList(
+            self.bundle_id, self.user, kCFPreferencesCurrentHost)
+        if keys is not None:
+            for i in keys:
+                yield i
+
+    def __contains__(self, pref_name):
+        pref_value = CFPreferencesCopyAppValue(pref_name, self.bundle_id)
+        return pref_value is not None
+
+    def __getitem__(self, pref_name):
+        return CFPreferencesCopyAppValue(pref_name, self.bundle_id)
+
+    def __setitem__(self, pref_name, pref_value):
+        CFPreferencesSetValue(
+            pref_name, pref_value, self.bundle_id, self.user,
+            kCFPreferencesCurrentHost)
+        CFPreferencesAppSynchronize(self.bundle_id)
+
+    def __delitem__(self, pref_name):
+        self.__setitem__(pref_name, None)
+
+    def __repr__(self):
+        return '<%s %s>' % (self.__class__.__name__, self.bundle_id)
+
+    def get(self, pref_name, default=None):
+        if pref_name not in self:
+            return default
+        else:
+            return self.__getitem__(pref_name)
+
+    def set(self, pref_name, pref_value):
+        self.__setitem__(pref_name, pref_value)
+
+
+class ServerFavoritesPrefs(Preferences):
+    def __init__(self):
+        self.bundle_id = PLIST_PATH
+
+
+class VNCFavorites(object):
+    def __init__(self):
+        self.prefs = ServerFavoritesPrefs()
+        self._servers = None
+
+    @property
+    def servers(self):
+        if not self._servers:
+            self._servers = self._get_servers()
+        return self._servers
+
+    def _get_servers(self):
+        return [
+            dict(name=server['Name'], url=server['URL'])
+            for server in self.prefs.get('favoriteservers')['CustomListItems']
+        ]
+
+if __name__ == '__main__':
+    v = VNCFavorites()
+    toXML = toXML()
+
+    for server in v.servers:
+        toXML.add_item(
+            title=server['name'],
+            subtitle="URL {url}".format(**server),
+            arg=server['url']
+        )
+    print toXML
No results found