U 楡ck#@sdZddlmZmZmZddlmZddlmZm Z ddl m Z ddl m Z dZGdd d ZGd d d ZGd d d ZGdddZddZGdddeeeZdS)z8Contains implementations of database retrieveing objects)join LazyMixin hex_to_bin) force_text) BadObjectAmbiguousObjectName)chain)reduce) ObjectDBR ObjectDBW FileDBBase CompoundDB CachingDBc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)r zkDefines an interface for object database lookup. Objects are identified either by their 20 byte bin shacCs|jSN)Zhas_objselfshar1/tmp/pip-unpacked-wheel-jjg5dubb/gitdb/db/base.py __contains__szObjectDBR.__contains__cCs tddS)z Whether the object identified by the given 20 bytes binary sha is contained in the database :return: True if the object identified by the given 20 bytes binary sha is contained in the databaseTo be implemented in subclassNNotImplementedErrorrrrr has_object"szObjectDBR.has_objectcCs tddS)zW :return: OInfo instance :param sha: bytes binary sha :raise BadObject:rNrrrrrinfo+szObjectDBR.infocCs tddS)z[:return: OStream instance :param sha: 20 bytes binary sha :raise BadObject:rNrrrrrstream1szObjectDBR.streamcCs tdS)z+:return: amount of objects in this databaseNrrrrrsize7szObjectDBR.sizecCs tdS)zGReturn iterator yielding 20 byte shas for all objects in this data baseNrrrrrsha_iter;szObjectDBR.sha_iterN) __name__ __module__ __qualname____doc__rrrrrrrrrrr s r c@s0eZdZdZddZddZddZdd Zd S) r z6Defines an interface to create objects in the databasecOs d|_dSrZ_ostream)rargskwargsrrr__init__FszObjectDBW.__init__cCs|j}||_|S)ab Adjusts the stream to which all data should be sent when storing new objects :param stream: if not None, the stream to use, if None the default stream will be used. :return: previously installed stream, or None if there was no override :raise TypeError: if the stream doesn't have the supported functionalityr#)rrZcstreamrrr set_ostreamJszObjectDBW.set_ostreamcCs|jS)z Return the output stream :return: overridden output stream this instance will write to, or None if it will write to the default streamr#rrrrostreamVszObjectDBW.ostreamcCs tddS)a Create a new object in the database :return: the input istream object with its sha set to its corresponding value :param istream: IStream compatible instance. If its sha is already set to a value, the object will just be stored in the our database format, in which case the input stream is expected to be in object format ( header + contents ). :raise IOError: if data could not be writtenrNr)rZistreamrrrstore^s zObjectDBW.storeN)rr r!r"r&r'r(r)rrrrr Bs  r cs0eZdZdZfddZddZddZZS)r z}Provides basic facilities to retrieve files of interest, including caching facilities to help mapping hexsha's to objectscst||_dS)aYInitialize this instance to look for its files at the given root path All subsequent operations will be relative to this path :raise InvalidDBRoot: **Note:** The base will not perform any accessablity checking as the base might not yet be accessible, but become accessible before the first access.N)superr& _root_path)r root_path __class__rrr&qs zFileDBBase.__init__cCs|jS)z':return: path at which this db operates)r+rrrrr,|szFileDBBase.root_pathcCst|jt|S)z~ :return: the given relative path relative to our database root, allowing to pontentially access datafiles)rr+r)rZ rela_pathrrrdb_pathszFileDBBase.db_path)rr r!r"r&r,r/ __classcell__rrr-rr ls r c@seZdZdZdddZdS)rz/A database which uses caches to speed-up accessFcCsdS)ar Call this method if the underlying data changed to trigger an update of the internal caching structures. :param force: if True, the update must be performed. Otherwise the implementation may decide not to perform an update if it thinks nothing has changed. :return: True if an update was performed as something change indeedNr)rforcerrr update_cacheszCachingDB.update_cacheN)F)rr r!r"r2rrrrrsrcCsTt|trF|}|dd|Ddd|DD]}t||q4n ||dS)zfFill output list with database from db, in order. Deals with Loose, Packed and compound databases.css|]}t|ts|VqdSr isinstancer .0dbrrr s z'_databases_recursive..css|]}t|tr|VqdSrr3r5rrrr8s N)r4r databasesextend_databases_recursiveappend)ZdatabaseoutputZdbsZcdbrrrr;s  r;csjeZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ dddZ ddZ ZS)r zA database which delegates calls to sub-databases. Databases are stored in the lazy-loaded _dbs attribute. Define _set_cache_ to update it with your databasescs4|dkrt|_n|dkr$t|_n t|dS)N_dbs _db_cache)listr>dictr?r* _set_cache_)rattrr-rrrBs   zCompoundDB._set_cache_cCsVz |j|WStk r YnX|jD] }||r(||j|<|Sq(t|dS)zL:return: database containing the given 20 byte sha :raise BadObject:N)r?KeyErrorr>rr)rrr7rrr _db_querys     zCompoundDB._db_querycCs,z||WdStk r&YdSXdS)NTF)rErrrrrrs  zCompoundDB.has_objectcCs|||Sr)rErrrrrrszCompoundDB.infocCs|||Sr)rErrrrrrszCompoundDB.streamcCstdddd|jDdS)z.:return: total size of all contained databasescSs||Srr)xyrrrz!CompoundDB.size..css|]}|VqdSr)rr5rrrr8sz"CompoundDB.size..r)r r>rrrrrszCompoundDB.sizecCstdd|jDS)Ncss|]}|VqdSr)rr5rrrr8sz&CompoundDB.sha_iter..)rr>rrrrrszCompoundDB.sha_itercCs t|jS)z7:return: tuple of database instances we use for lookups)tupler>rrrrr9szCompoundDB.databasesFcCs6|jd}|jD]}t|tr|||O}q|S)NF)r?clearr>r4rr2)rr1statr7rrrr2s    zCompoundDB.update_cachec Cst}t||t|}t|}|ddkr:t|d}nt|}d}|D]d}d}z&t|drj||}n |||}Wntk rYqJYnX|rJ|r||krt ||}qJ|st||S)z :return: 20 byte binary sha1 from the given less-than-40 byte hexsha (bytes or str) :param partial_hexsha: hexsha with less than 40 byte :raise AmbiguousObjectName: r0Npartial_to_complete_sha_hex) r@r;rlenrhasattrrOZpartial_to_complete_sharr)rZpartial_hexshar9Zlen_partial_hexshaZpartial_binsha candidater7Z full_bin_sharrrrOs.      z&CompoundDB.partial_to_complete_sha_hex)F)rr r!r"rBrErrrrrr9r2rOr0rrr-rr s  r N)r"Z gitdb.utilrrrZgitdb.utils.encodingrZ gitdb.excrr itertoolsr functoolsr __all__r r r rr;r rrrrs   )*