o j_ @sNdZddlmZmZmZddlmZddlmZm Z ddl m Z ddl m Z mZmZddlmZddlmZmZmZmZmZed Zed Zd Zd ZGd d d eZ  d"dedededee def ddZdedeefddZ d#dedee defddZ!dedefddZ"dededefddZ#dedeeefdeefd d!Z$dS)$arCollection is a place where secret items are stored. Normally, only the default collection should be used, but this module allows to use any registered collection. Use :func:`get_default_collection` to get the default collection (and create it, if necessary). Collections are usually automatically unlocked when user logs in, but collections can also be locked and unlocked using :meth:`Collection.lock` and :meth:`Collection.unlock` methods (unlocking requires showing the unlocking prompt to user and can be synchronous or asynchronous). Creating new items and editing existing ones is possible only in unlocked collection.)DictIteratorOptional)DBusConnection) SS_PREFIXSS_PATH)Session)LockedExceptionItemNotFoundExceptionPromptDismissedException)Item)DBusAddressWrapper exec_prompt format_secret open_sessionunlock_objects CollectionZServicez(/org/freedesktop/secrets/aliases/defaultz+/org/freedesktop/secrets/collection/sessionc @seZdZdZedfdededeeddfddZ de fd d Z d$d d Z de fd dZ d$ddZd$ddZdeefddZdeeefdeefddZdefddZdeddfddZ  d%dedeeefded e d!edef d"d#ZdS)&rzRepresents a collection.N connectioncollection_pathsessionreturncCs0||_||_||_t|t||_|jddS)NLabel)rrrr COLLECTION_IFACE _collection get_property)selfrrrr:/usr/lib/python3/dist-packages/secretstorage/collection.py__init__$szCollection.__init__cCst|jdS)zDReturns :const:`True` if item is locked, otherwise :const:`False`.ZLocked)boolrrrrrr is_locked.szCollection.is_lockedcCs|rtddS)zSIf collection is locked, raises :exc:`~secretstorage.exceptions.LockedException`.zCollection is locked!N)r!r r rrrensure_not_locked3szCollection.ensure_not_lockedcCst|j|jgS)aRequests unlocking the collection. Returns a boolean representing whether the prompt has been dismissed; that means :const:`False` on successful unlocking and :const:`True` if it has been dismissed. .. versionchanged:: 3.0 No longer accepts the ``callback`` argument. )rrrr rrrunlock9s zCollection.unlockcCs$ttt|j}|dd|jgdS)zLocks the collection.ZLockZaoN)r r SERVICE_IFACErcallr)rservicerrrlockEszCollection.lockcCsD||jdd\}|dkrt|j|\}}|r tddSdS)z/Deletes the collection and all items inside it.ZDelete/Prompt dismissed.N)r"rr%rrr )rprompt dismissedZ_resultrrrdeleteJszCollection.deleteccs*|jdD] }t|j||jVqdS)z3Returns a generator of all items in the collection.ZItemsN)rrr rr)r item_pathrrr get_all_itemsSszCollection.get_all_items attributesccs4|jdd|\}|D] }t|j||jVq dS)z^Returns a generator of items with the given attributes. `attributes` should be a dictionary. SearchItemsa{ss}N)rr%r rr)rr0resultr.rrr search_itemsXs zCollection.search_itemscCs|jd}t|ts J|S)zReturns the collection label.r)rr isinstancestrrlabelrrr get_label_s zCollection.get_labelr8cCs||jdd|dS)z!Sets collection label to `label`.rsN)r"rZ set_propertyr7rrr set_labeleszCollection.set_labelF text/plainsecretreplace content_typec Csl||js t|j|_t|j||}tdd|ftdd|fi}|jdd|||\}} t|j||jS)acCreates a new :class:`~secretstorage.item.Item` with given `label` (unicode string), `attributes` (dictionary) and `secret` (bytestring). If `replace` is :const:`True`, replaces the existing item with the same attributes. If `content_type` is given, also sets the content type of the secret (``text/plain`` by default). Returns the created item.z Item.Labelr:zItem.Attributesr2Z CreateItemza{sv}(oayays)b) r"rrrrrrr%r ) rr8r0r=r>r?Z_secret propertiesZnew_itemr+rrr create_itemjs     zCollection.create_item)rN)Fr<)__name__ __module__ __qualname____doc__DEFAULT_COLLECTIONrr6rrrrr!r"r#r'r-rr r/rr4r9r;bytesrArrrrr!s>    r(Nrr8aliasrrc Cs|st|}tdd|fi}ttt|}|dd||\}}t|dkr+t|||dSt||\}} |r8t d| \} }| dksBJt|||dS) zCreates a new :class:`Collection` with the given `label` and `alias` and returns it. This action requires prompting. :raises: :exc:`~secretstorage.exceptions.PromptDismissedException` if the prompt is dismissed. zCollection.Labelr:ZCreateCollectionza{sv}s)rr*o) rrr rr$r%lenrrr ) rr8rHrr@r&rr+r,r3Z signaturerrrcreate_collections   rLccs.ttt|}|dD]}t||Vq dS)z1Returns a generator of all available collections.Z CollectionsN)r rr$rr)rr&rrrrget_all_collectionss  rMcCs,zt|WStyt|dd|YSw)zAReturns the default collection. If it doesn't exist, creates it.ZDefaultdefault)rr rL)rrrrrget_default_collections  rOcCs^zt|WStyYnwzt|tWStyYnwtt|}|r+|dStd)zReturns any collection, in the following order of preference: - The default collection; - The "session" collection (usually temporary); - The first collection in the collections list.rzNo collections found.)rr SESSION_COLLECTIONlistrM)r collectionsrrrget_any_collections     rScCs:ttt|}|dd|\}t|dkrtdt||S)zReturns the collection with the given `alias`. If there is no such collection, raises :exc:`~secretstorage.exceptions.ItemNotFoundException`.Z ReadAliasr:rIzNo collection with such alias.)r rr$r%rKr r)rrHr&rrrrget_collection_by_aliass   rTr0ccs>ttt|}|dd|\}}||D]}t||VqdS)zpReturns a generator of items in all collections with the given attributes. `attributes` should be a dictionary.r1r2N)r rr$r%r )rr0r&lockedZunlockedr.rrrr4s   r4)r(N)N)%rEtypingrrrZjeepney.io.blockingrZsecretstorage.definesrrZsecretstorage.dhcryptorZsecretstorage.exceptionsr r r Zsecretstorage.itemr Zsecretstorage.utilr rrrrrr$rFrPobjectrr6rLrMrOrSrTr4rrrrsL   ^