o il% @sDdZddlmZddlmZddlmZmZddlm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZmZed Zed Zd Zd ZGd d d Z  d"dededede dBdef ddZdedeefddZ d#dede dBdefddZdedefddZdededefddZ dede!eefdeefd d!Z"dS)$azCollection 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 blocks until user accepts or declines it). Creating new items and editing existing ones is possible only in unlocked collections. )Iterator)DBusConnection)SS_PATH SS_PREFIX)Session)ItemNotFoundExceptionLockedExceptionPromptDismissedException)Item)DBusAddressWrapper exec_prompt format_secret open_sessionunlock_objects CollectionServicez(/org/freedesktop/secrets/aliases/defaultz+/org/freedesktop/secrets/collection/sessionc @seZdZdZedfdedededBddfddZde fd d Z d&d d Z de fd dZ d&ddZ d&ddZdeefddZdeeefdeefddZdefddZdeddfddZ  d'dedeeefded e d!edef d"d#Zdefd$d%ZdS)(rzRepresents a collection.N connectioncollection_pathsessionreturncCs0||_||_||_t|t||_|jddS)NLabel)rrrr COLLECTION_IFACE _collection get_property)selfrrrr^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/secretstorage/collection.py__init__0szCollection.__init__cCst|jdS)zJReturns :const:`True` if item is locked, otherwise :const:`False`.Locked)boolrrrrrr is_locked:szCollection.is_lockedcCs|rtddS)zYIf collection is locked, raises :exc:`~secretstorage.exceptions.LockedException`.zCollection is locked!N)r!rr rrrensure_not_locked?szCollection.ensure_not_lockedcCst|j|jgS)aARequests 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 rrrunlockEs zCollection.unlockcCs$ttt|j}|dd|jgdS)zLocks the collection.LockaoN)r r SERVICE_IFACErcallr)rservicerrrlockQszCollection.lockcCsD||jdd\}|dkrt|j|\}}|r tddSdS)z/Deletes the collection and all items inside it.Delete/Prompt dismissed.N)r"rr'r rr )rprompt dismissed_resultrrrdeleteVszCollection.deleteccs*|jdD] }t|j||jVqdS)z3Returns a generator of all items in the collection.ItemsN)rrr rr)r item_pathrrr get_all_items_szCollection.get_all_items attributesccs4|jdd|\}|D] }t|j||jVq dS)zdReturns a generator of items with the given attributes. `attributes` should be a dictionary. SearchItemsa{ss}N)rr'r rr)rr5resultr3rrr search_itemsds zCollection.search_itemscCs|jd}t|ts J|S)zReturns the collection label.r)rr isinstancestrrlabelrrr get_labelks zCollection.get_labelr=cCs||jdd|dS)z!Sets collection label to `label`.rsN)r"r set_propertyr<rrr set_labelqszCollection.set_labelF text/plainsecretreplace content_typec Cs||js t|j|_t|j||}tdd|ftdd|fi}|jdd|||\}} t|dkr)r>rr rrr__repr__szCollection.__repr__)rN)FrB)__name__ __module__ __qualname____doc__DEFAULT_COLLECTIONrr;rrrr!r"r#r)r1rr r4dictr9r>rAbytesrMrOrrrrr-s@     !r+Nrr=aliasrrc 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?CreateCollectionza{sv}srG)rr-rH) rrr rr&r'rIrr r ) rr=rWrrKr(rr.r/r8rLrrrcreate_collections   rYccs.ttt|}|dD]}t||Vq dS)z1Returns a generator of all available collections. CollectionsN)r rr&rr)rr(rrrrget_all_collectionss  r[cCs,zt|WStyt|dd|YSw)zDReturns the default collection. If it doesn't exist, creates it.Defaultdefault)rrrY)rrrrrget_default_collections   r^cCs^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.)rrSESSION_COLLECTIONlistr[)r collectionsrrrget_any_collections     rbcCs:ttt|}|dd|\}t|dkrtdt||S)zReturns the collection with the given `alias`. If there is no such collection, raises :exc:`~secretstorage.exceptions.ItemNotFoundException`. ReadAliasr?rGzNo collection with such alias.)r rr&r'rIrr)rrWr(rrrrget_collection_by_aliass   rdr5ccs>ttt|}|dd|\}}||D]}t||VqdS)zsReturns a generator of items in all collections with the given attributes. `attributes` should be a dictionary.r6r7N)r rr&r'r )rr5r(lockedunlockedr3rrrr9s   r9)r+N)N)#rScollections.abcrjeepney.io.blockingrsecretstorage.definesrrsecretstorage.dhcryptorsecretstorage.exceptionsrrr secretstorage.itemr secretstorage.utilr r r rrrr&rTr_rr;rYr[r^rbrdrUr9rrrrsL   n