mirror of
https://github.com/EvolutionAPI/evolution-client-python.git
synced 2026-02-04 13:56:23 -06:00
initial commit
This commit is contained in:
@@ -0,0 +1,155 @@
|
||||
import logging
|
||||
|
||||
from .. import backend
|
||||
from ..backend import KeyringBackend
|
||||
from ..compat import properties
|
||||
from ..credentials import SimpleCredential
|
||||
from ..errors import (
|
||||
KeyringLocked,
|
||||
PasswordDeleteError,
|
||||
PasswordSetError,
|
||||
)
|
||||
|
||||
available = False
|
||||
try:
|
||||
import gi
|
||||
from gi.repository import Gio, GLib
|
||||
|
||||
gi.require_version('Secret', '1')
|
||||
from gi.repository import Secret
|
||||
|
||||
available = True
|
||||
except (AttributeError, ImportError, ValueError):
|
||||
pass
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Keyring(backend.SchemeSelectable, KeyringBackend):
|
||||
"""libsecret Keyring"""
|
||||
|
||||
appid = 'Python keyring library'
|
||||
|
||||
@property
|
||||
def schema(self):
|
||||
return Secret.Schema.new(
|
||||
"org.freedesktop.Secret.Generic",
|
||||
Secret.SchemaFlags.NONE,
|
||||
self._query(
|
||||
Secret.SchemaAttributeType.STRING,
|
||||
Secret.SchemaAttributeType.STRING,
|
||||
application=Secret.SchemaAttributeType.STRING,
|
||||
),
|
||||
)
|
||||
|
||||
@properties.NonDataProperty
|
||||
def collection(self):
|
||||
return Secret.COLLECTION_DEFAULT
|
||||
|
||||
@properties.classproperty
|
||||
def priority(cls) -> float:
|
||||
if not available:
|
||||
raise RuntimeError("libsecret required")
|
||||
|
||||
# Make sure there is actually a secret service running
|
||||
try:
|
||||
Secret.Service.get_sync(Secret.ServiceFlags.OPEN_SESSION, None)
|
||||
except GLib.Error as error:
|
||||
raise RuntimeError("Can't open a session to the secret service") from error
|
||||
|
||||
return 4.8
|
||||
|
||||
def get_password(self, service, username):
|
||||
"""Get password of the username for the service"""
|
||||
attributes = self._query(service, username, application=self.appid)
|
||||
try:
|
||||
items = Secret.password_search_sync(
|
||||
self.schema, attributes, Secret.SearchFlags.UNLOCK, None
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('g-io-error-quark')
|
||||
if error.matches(quark, Gio.IOErrorEnum.FAILED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
for item in items:
|
||||
try:
|
||||
return item.retrieve_secret_sync().get_text()
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('secret-error')
|
||||
if error.matches(quark, Secret.Error.IS_LOCKED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
|
||||
def set_password(self, service, username, password):
|
||||
"""Set password for the username of the service"""
|
||||
attributes = self._query(service, username, application=self.appid)
|
||||
label = f"Password for '{username}' on '{service}'"
|
||||
try:
|
||||
stored = Secret.password_store_sync(
|
||||
self.schema, attributes, self.collection, label, password, None
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('secret-error')
|
||||
if error.matches(quark, Secret.Error.IS_LOCKED):
|
||||
raise KeyringLocked("Failed to unlock the collection!") from error
|
||||
quark = GLib.quark_try_string('g-io-error-quark')
|
||||
if error.matches(quark, Gio.IOErrorEnum.FAILED):
|
||||
raise KeyringLocked("Failed to unlock the collection!") from error
|
||||
raise
|
||||
if not stored:
|
||||
raise PasswordSetError("Failed to store password!")
|
||||
|
||||
def delete_password(self, service, username):
|
||||
"""Delete the stored password (only the first one)"""
|
||||
attributes = self._query(service, username, application=self.appid)
|
||||
try:
|
||||
items = Secret.password_search_sync(
|
||||
self.schema, attributes, Secret.SearchFlags.UNLOCK, None
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('g-io-error-quark')
|
||||
if error.matches(quark, Gio.IOErrorEnum.FAILED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
for item in items:
|
||||
try:
|
||||
removed = Secret.password_clear_sync(
|
||||
self.schema, item.get_attributes(), None
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('secret-error')
|
||||
if error.matches(quark, Secret.Error.IS_LOCKED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
return removed
|
||||
raise PasswordDeleteError("No such password!")
|
||||
|
||||
def get_credential(self, service, username):
|
||||
"""Get the first username and password for a service.
|
||||
Return a Credential instance
|
||||
|
||||
The username can be omitted, but if there is one, it will use get_password
|
||||
and return a SimpleCredential containing the username and password
|
||||
Otherwise, it will return the first username and password combo that it finds.
|
||||
"""
|
||||
query = self._query(service, username)
|
||||
try:
|
||||
items = Secret.password_search_sync(
|
||||
self.schema, query, Secret.SearchFlags.UNLOCK, None
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('g-io-error-quark')
|
||||
if error.matches(quark, Gio.IOErrorEnum.FAILED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
for item in items:
|
||||
username = item.get_attributes().get("username")
|
||||
try:
|
||||
return SimpleCredential(
|
||||
username, item.retrieve_secret_sync().get_text()
|
||||
)
|
||||
except GLib.Error as error:
|
||||
quark = GLib.quark_try_string('secret-error')
|
||||
if error.matches(quark, Secret.Error.IS_LOCKED):
|
||||
raise KeyringLocked('Failed to unlock the item!') from error
|
||||
raise
|
||||
Reference in New Issue
Block a user