o he#@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 shacC|jSN)has_objselfsharA/var/www/vscode/kcb/lib/python3.10/site-packages/gitdb/db/base.py __contains__szObjectDBR.__contains__cCtd)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 subclassNotImplementedErrorrrrr has_object"szObjectDBR.has_objectcCr)zW :return: OInfo instance :param sha: bytes binary sha :raise BadObject:rrrrrrinfo+zObjectDBR.infocCr)z[:return: OStream instance :param sha: 20 bytes binary sha :raise BadObject:rrrrrrstream1rzObjectDBR.streamcCt)z+:return: amount of objects in this databaserrrrrsize7zObjectDBR.sizecCr )zGReturn iterator yielding 20 byte shas for all objects in this data baserr!rrrsha_iter;r#zObjectDBR.sha_iterN) __name__ __module__ __qualname____doc__rrrrr"r$rrrrr s  r c@s0eZdZdZddZddZddZdd Zd S) r z6Defines an interface to create objects in the databasecOs d|_dSr_ostream)rargskwargsrrr__init__Fs zObjectDBW.__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))rrcstreamrrr set_ostreamJszObjectDBW.set_ostreamcCr)z Return the output stream :return: overridden output stream this instance will write to, or None if it will write to the default streamr)r!rrrostreamVszObjectDBW.ostreamcCr)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 writtenrr)ristreamrrrstore^s zObjectDBW.storeN)r%r&r'r(r-r/r0r2rrrrr 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__cCr)z':return: path at which this db operates)r4r!rrrr5|r#zFileDBBase.root_pathcCst|jt|S)z~ :return: the given relative path relative to our database root, allowing to pontentially access datafiles)rr4r)r rela_pathrrrdb_pathszFileDBBase.db_path)r%r&r'r(r-r5r9 __classcell__rrr6rr 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_cacheNF)r%r&r'r(r<rrrrrsrcCsVt|tr$|}|dd|Ddd|DD]}t||qdS||dS)zfFill output list with database from db, in order. Deals with Loose, Packed and compound databases.css|] }t|ts|VqdSr isinstancer .0dbrrr z'_databases_recursive..css|] }t|tr|VqdSrr>r@rrrrCrDN)r?r databasesextend_databases_recursiveappend)databaseoutputdbscdbrrrrGs  rGcsjeZdZdZfddZddZddZdd 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 databasescs8|dkr t|_dS|dkrt|_dSt|dS)N_dbs _db_cache)listrMdictrNr3 _set_cache_)rattrr6rrrQs   zCompoundDB._set_cache_cCsPz|j|WStyYnw|jD]}||r#||j|<|Sqt|)zL:return: database containing the given 20 byte sha :raise BadObject:)rNKeyErrorrMrr)rrrBrrr _db_querys     zCompoundDB._db_querycCs&z||WdStyYdSw)NTF)rTrrrrrrs   zCompoundDB.has_objectcC|||Sr)rTrrrrrrzCompoundDB.infocCrUr)rTrrrrrrrVzCompoundDB.streamcCstdddd|jDdS)z.:return: total size of all contained databasescSs||Srr)xyrrrsz!CompoundDB.size..cs|]}|VqdSr)r"r@rrrrCz"CompoundDB.size..r)r rMr!rrrr"szCompoundDB.sizecCstdd|jDS)NcsrZr)r$r@rrrrCr[z&CompoundDB.sha_iter..)rrMr!rrrr$szCompoundDB.sha_itercCs t|jS)z7:return: tuple of database instances we use for lookups)tuplerMr!rrrrEs zCompoundDB.databasesFcCs6|jd}|jD]}t|tr|||O}q |S)NF)rNclearrMr?rr<)rr;statrBrrrr<s   zCompoundDB.update_cachec Cst}t||t|}t|}|ddkrt|d}nt|}d}|D]/}d}zt|dr5||}n|||}Wn tyEYq%w|rT|rR||krRt ||}q%|s[t||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) rOrGrlenrhasattrrapartial_to_complete_sharr)rpartial_hexsharElen_partial_hexshapartial_binsha candidaterB full_bin_sharrrras4       z&CompoundDB.partial_to_complete_sha_hexr=)r%r&r'r(rQrTrrrr"r$rEr<rar:rrr6rr s   r N)r( gitdb.utilrrrgitdb.utils.encodingr gitdb.excrr itertoolsr functoolsr __all__r r r rrGr rrrrs   )*