Sea.UdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlZedaedad Zd Zej&eZGd d eZ ddlZej2ZGddeZGddeZGddeZgdZddgZ dddddddddd d!d"d#d$ Z!d%d&d'd(d)d*d+d,d-d.d/d0d1d2 Z"d3d4d5d6d7d8d9d:ejFd;g Z$dd?d@Z%dAa&ejNZ(ee(jRZ*ejVdBZ,ejVdCZ-idDddEdFdGdFdHdFdIdFdJdFdKdFdLdFdMddNddOdFdPdFdQddRddSddTddUddVddVdFdVdVdVdddVdddW Z.iZ/e0e1dX<iZ2e0e1dY<iZ3e0e1dZ<iZ4e0e1d[<dmd\e5d]e5fd^Z6dnd_e7d]dfd`Z8dodaZ9ejte9dbZ;dcedodfZ?dndge7d]dfdhZ@diZAdjZBdke5d]e5fdlZCy#e$rd ZYpwxYw)pa Copyright (c) 2019-2022, Ian Santopietro All rights reserved. This file is part of RepoLib. RepoLib is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. RepoLib is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with RepoLib. If not, see . N)Enum)Path)urlparse)requesterror/etc/apt/sources.list.dz/etc/apt/keyrings/Fz7http://keyserver.ubuntu.com/pks/lookup?op=get&search=0xc(eZdZdZddfd ZxZS) RepoErrorz Exception from this module.)codec2t||i|||_y)zException with a source object Arguments: code (:obj:`int`, optional, default=1): Exception error code. N)super__init__r )selfr argskwargs __class__s ./usr/lib/python3/dist-packages/repolib/util.pyrzRepoError.__init__-s $)&) )__name__ __module__ __qualname____doc__r __classcell__)rs@rr r *s&#$rr linuxceZdZdZdZdZy) SourceFormatzEnum of SourceFile FormatssourceslistN)rrrrDEFAULTLEGACYrrrr<s$G Frrc$eZdZdZdZdZdefdZy) SourceTypezEnum of repository typesdebdeb-srcreturnch|j}|jdd}|jdd}|S)z3Used for getting a version of the format for identsr&sourcer%binary)valuereplace)ridents rr-zSourceType.identFs2::, i2 eX. rN)rrrrBINARY SOURCECODEstrr-r"rrr$r$As" FJsrr$ceZdZdZdZdZdZy)AptSourceEnabledzC Helper Enum to translate between bool data and the Deb822 format. yesnoc$|jdk(ryy)z# Return a bool based on the value. r3TF)r+)rs rget_boolzAptSourceEnabled.get_boolRs :: rN)rrrrTRUEFALSEr6r"rrr2r2MsM D Err2)zX-Repolib-Name:z X-Repolib-ID:zX-Repolib-Default-Mirror:zX-Repolib-CommentX-Repolib-PrefszEnabled:zTypes:zURIs:zSuites:z Components:zArchitectures:z Languages:zTargets:zPDiffs:zBy-Hash:zAllow-Insecure:z Allow-Weak:zAllow-Downgrade-To-Insecure:zTrusted:z Signed-By:zCheck-Valid-Until:zValid-Until-Min:zValid-Until-Max:r9 X-Repolib-ID Architectures LanguagesTargetsPDiffsBy-HashAllow-Insecure Allow-WeakAllow-Downgrade-To-InsecureTrusted Signed-ByCheck-Valid-UntilValid-Until-MinValid-Until-Max) archlangtargetpdiffsby-hashallow-insecure allow-weakallow-downgrade-to-insecuretrusted signed-bycheck-valid-untilvalid-until-minvalid-until-maxrHrIrJrKrLrMrNrOrPrQrRrSrT) r;r<r=r>r?r@rArBrCrDrErFrGTTruetrueYesr3YESyYr zName: zIdent: z Comments: zDefault Mirror: )zX-Repolib-Name: zX-Repolib-ID: zX-Repolib-Comments: zX-Repolib-Default-Mirror:  z[^@.+]\[([^[]+.+)\]\ z\w+:(\/?\/?)[^\s]+!@-#$%^&*)(+=[\]{}_) |?/.<>,`~ :;rfileskeyserrorsnamer'c,|jtS)z Clean up a string intended for a filename. Arguments: name (str): The prospective name to scrub. Returns: str The cleaned-up name. ) translate CLEAN_CHARS)r~s rscrub_filenamers >>+ &&rtestingctj}|sdadayt |j }|dz dz dz a|dz dz d z ay) zSets Repolib in testing mode where changes will not be saved. Arguments: testing(bool): Whether testing mode should be enabled or disabled (Defaul: True) z/usr/share/keyringsrNusrsharekeyringsetcaptzsources.list.d)tempfileTemporaryDirectoryKEYS_DIR SOURCES_DIRrr~)rtesting_tempdir testing_roots r set_testingrs[113O (/ ,,-Le#g- :H&.1AAKrc,tjy)zClean up our tempdirN) _KEYS_TEMPDIRcleanupr"rr_cleanup_temspsrsrcptj}|jdd}|jy)Nzorg.pop_os.repolibz/Repo)dbus SystemBus get_objectexit)busprivileged_objects r dbus_quitrs, .. C';WEr excl_keysc|D]}||vr||vr||||k7ryy|D]}||vr||vr||||k7ryyy)aCompare two sources based on arbitrary criteria. This looks at a given list of keys, and if the given keys between the two given sources are identical, returns True. Arguments: source1, source2(Source): The two sources to compare excl_keys([str]): Any keys to exclude from the comparison Returns: bool `True` if the sources are identical, otherwise `False`. FTr")source1source2rkeys rcompare_sourcesr s  )   '>s|ws|+   )   '>s|ws|+  rci}|D].}||vr||vr||||k(r||||f||<||df||<0|D]!}||vr||vr ||||k(rd||f||<#|S)a]Find key-value pairs which differ between two sources. Arguments: source1, source2(Source): The two sources to compare excl_keys([str]): Any keys to exclude from the comparison Returns: dict{'key': ('source1[key]','source2[key]')} The dictionary of different keys, with the key values from each source. r")rrrdiffering_keysrs rfind_differences_sourcesr,sN1 )   '>s|ws|+#*3<">N3 &s|R0s11 )   '>s|ws|+!73<0s 1 rc|D]}|dvr||vs ||xxd||z cc<!|D]}|dvr||vs ||xxd||z cc<!|D]V}||jj}g}|D]}||vs|j|dj|||<X|D]V}||jj}g}|D]}||vs|j|dj|||<Xy)zCombine the data in two sources into one. Arguments: source1(Source): The source to be merged into source2(Source): The source to merge from )zX-Repolib-Namer:EnabledTypes N)stripsplitappendjoin)rrrvalsnewvalsvals rcombine_sourcesrJsL/ H H  '> CLa ~. .L / / H H  '> CLa ~. .L /)s|!!#))+ $C'!s# $xx( ))s|!!#))+ $C'!s# $xx( )renabledc|rdayday)zEasy helper to enable/disable pretty-printing for object reprs. Can also be used as an easy way to reset to defaults. Arguments: enabled(bool): Whether or not Pretty Printing should be enabled r[rN) PRETTY_PRINT)rs rprettyprint_enablerms  rc t|}|jsy|jdk(ry|jr!t|j|jgS|jr!t|j|jgSy#YyxYw)z Validate a url and tell if it's good or not. Arguments: url (str): The URL to validate. Returns: `True` if `url` is not malformed, otherwise `False`. Fzx-repolib-name)rschemenetlocallpath)urlresults r url_validatorr{sw#}} ==, , == v}}56 6 [[ v{{34 4sBB,B,BB cd}|jdr$d}|jdd}|j}|jdr%|j}|D]}t |syy |jdrd|vryy |j dryt |jd k(r |s t |Sy) z Basic checks to see if a given debline is valid or not. Arguments: valid (str): The line to validate. Returns: True if the line is valid, False otherwise. F#Trr%zppa:/z .flatpakrepor N) startswithr,rrrendswithlen)validcommentwordswords rvalidate_deblinersG  c2&    DT"    & ! %<  >>. ) u{{}  "7 ' 'rlinecp |jd}|j}|jds |S6)z Strips the leading #'s from the given line. Arguments: line (str): The line to strip. Returns: (str): The input line without any leading/trailing hashes or leading/trailing whitespace. r)rr)rs r strip_hashesrs8 zz#zz|s#  K r)r)T)r'N)Dratexitloggingrerenumrpathlibr urllib.parserurllibrrrrrTESTINGKEYSERVER_QUERY_URL getLoggerrlog Exceptionr distrocodenameDISTRO_CODENAME ImportErrorrr$r2 valid_keysoutput_skip_keys options_inmapoptions_outmapr7 true_valueskeys_maprrrr~TEMP_DIRcompile options_reuri_rerrdict__annotations__r{r|r}r0rboolrrregisterrrrrrrrrrr"rrrs}( !! ,- $ % Og!  %foo'O4   t  6  &#@,(( "&#@,(("        !("4   +++-  "" # RZZ0 1 ) *      !"#$           ; @ d Tt ' 'c 'BBdB,   !!!F$< )F   8@cc} Os$GG*)G*