examples: add update secrets example

examples/python/update-secrets.py

@@ -0,0 +1,121 @@
+#!/bin/env python
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+# Copyright (C) 2011 Red Hat, Inc.
+#
+
+#
+# The example shows how to update secrets in a connection by means of D-Bus
+# Update() method. The method replaces all previous settings with new ones
+# including possible secrets.
+# So, we get all settings using GetSettings() and then find out what secrets
+# are associated with the connection using GetSecrets(), ask for new secret 
+# values, and add them to the settings that we pass to Update().
+#
+
+import dbus
+import sys
+
+bus = dbus.SystemBus()
+
+def change_secrets_in_one_setting(proxy, config, setting_name):
+        # Add new secret values to the connection config
+    try:
+        # returns a dict of dicts mapping name::setting, where setting is a dict
+        # mapping key::value.  Each member of the 'setting' dict is a secret
+        secrets = proxy.GetSecrets(setting_name, [], False)
+
+        # Ask user for new secrets and put them into our connection config
+        for setting in secrets:
+            for key in secrets[setting]:
+                new_secret = raw_input ("Enter new secret for %s in %s: " % (key, setting))
+                config[setting_name][key] = new_secret
+    except Exception, e:
+        code = str(e).split(':')[0]
+        if code == "org.freedesktop.DBus.GLib.UnmappedError.NmSettingsInterfaceErrorQuark.Code5":
+            sys.exit("Not able to get secrets, run as root")
+        else:
+            pass
+
+def change_secrets(service_name, con_path, config):
+    # Get existing secrets; we grab the secrets for each type of connection
+    # (since there isn't a "get all secrets" call because most of the time
+    # you only need 'wifi' secrets or '802.1x' secrets, not everything) and
+    # set new values into the connection settings (config)
+    con_proxy = bus.get_object(service_name, con_path)
+    connection_secrets = dbus.Interface(con_proxy, "org.freedesktop.NetworkManagerSettings.Connection.Secrets")
+    change_secrets_in_one_setting(connection_secrets, config, '802-11-wireless')
+    change_secrets_in_one_setting(connection_secrets, config, '802-11-wireless-security')
+    change_secrets_in_one_setting(connection_secrets, config, '802-1x')
+    change_secrets_in_one_setting(connection_secrets, config, 'gsm')
+    change_secrets_in_one_setting(connection_secrets, config, 'cdma')
+    change_secrets_in_one_setting(connection_secrets, config, 'ppp')
+
+def find_connection(name, service_name):
+    # Ask the settings service for the list of connections it provides
+    global con_path
+    proxy = bus.get_object(service_name, "/org/freedesktop/NetworkManagerSettings")
+    settings = dbus.Interface(proxy, "org.freedesktop.NetworkManagerSettings")
+    connection_paths = settings.ListConnections()
+
+    # Get the settings and look for connection's name
+    for path in connection_paths:
+        con_proxy = bus.get_object(service_name, path)
+        connection = dbus.Interface(con_proxy, "org.freedesktop.NetworkManagerSettings.Connection")
+        config = connection.GetSettings()
+
+        # Find connection by the id
+	s_con = config['connection']
+        if name == s_con['id']:
+            con_path = path
+            return config
+
+    return None
+
+
+# Main part
+con_path = None
+
+if len(sys.argv) != 2 and len(sys.argv) != 3:
+    sys.exit("Usage: %s <connection name> [user|system]" % sys.argv[0])
+
+# Find the connection
+if len(sys.argv) == 3:
+    if sys.argv[2] == 'system':
+        setting_service = "org.freedesktop.NetworkManagerSystemSettings"
+    elif sys.argv[2] == 'user':
+        setting_service = "org.freedesktop.NetworkManagerUserSettings"
+    else:
+        sys.exit("Usage: %s <connection name> [user|system]" % sys.argv[0])
+    con = find_connection(sys.argv[1], setting_service)
+else:
+    setting_service = "org.freedesktop.NetworkManagerSystemSettings"
+    con = find_connection(sys.argv[1], setting_service)
+    if not con:
+        setting_service = "org.freedesktop.NetworkManagerUserSettings"
+        con = find_connection(sys.argv[1], setting_service)
+
+if con:
+    # Obtain new secrets and put then into connection dict 
+    change_secrets(setting_service, con_path, con)
+
+    # Change the connection with Update()
+    proxy = bus.get_object(setting_service, con_path)
+    settings = dbus.Interface(proxy, "org.freedesktop.NetworkManagerSettings.Connection")
+    settings.Update(con)
+else:
+    sys.exit("No connection '%s' found" % sys.argv[1])
+