Uh oh!
There was an error while loading.Please reload this page.
- Notifications
You must be signed in to change notification settings - Fork966
Expand file tree
/
Copy pathconfig.py
More file actions
963 lines (777 loc) · 35 KB
/
config.py
File metadata and controls
963 lines (777 loc) · 35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
#
# This module is part of GitPython and is released under the
# 3-Clause BSD License: https://opensource.org/license/bsd-3-clause/
"""Parser for reading and writing configuration files."""
__all__= ["GitConfigParser","SectionConstraint"]
importabc
importconfigparserascp
importfnmatch
fromfunctoolsimportwraps
importinspect
fromioimportBufferedReader,IOBase
importlogging
importos
importos.pathasosp
importre
importsys
fromgit.compatimportdefenc,force_text
fromgit.utilimportLockFile
# typing-------------------------------------------------------
fromtypingimport (
Any,
Callable,
Generic,
IO,
List,
Dict,
Sequence,
TYPE_CHECKING,
Tuple,
TypeVar,
Union,
cast,
)
fromgit.typesimportLit_config_levels,ConfigLevels_Tup,PathLike,assert_never,_T
ifTYPE_CHECKING:
fromioimportBytesIO
fromgit.repo.baseimportRepo
T_ConfigParser=TypeVar("T_ConfigParser",bound="GitConfigParser")
T_OMD_value=TypeVar("T_OMD_value",str,bytes,int,float,bool)
ifsys.version_info[:3]< (3,7,2):
# typing.Ordereddict not added until Python 3.7.2.
fromcollectionsimportOrderedDict
OrderedDict_OMD=OrderedDict
else:
fromtypingimportOrderedDict
OrderedDict_OMD=OrderedDict[str,List[T_OMD_value]]# type: ignore[assignment, misc]
# -------------------------------------------------------------
_logger=logging.getLogger(__name__)
CONFIG_LEVELS:ConfigLevels_Tup= ("system","user","global","repository")
"""The configuration level of a configuration file."""
CONDITIONAL_INCLUDE_REGEXP=re.compile(r"(?<=includeIf )\"(gitdir|gitdir/i|onbranch|hasconfig:remote\.\*\.url):(.+)\"")
"""Section pattern to detect conditional includes.
See: https://git-scm.com/docs/git-config#_conditional_includes
"""
classMetaParserBuilder(abc.ABCMeta):# noqa: B024
"""Utility class wrapping base-class methods into decorators that assure read-only
properties."""
def__new__(cls,name:str,bases:Tuple,clsdict:Dict[str,Any])->"MetaParserBuilder":
"""Equip all base-class methods with a needs_values decorator, and all non-const
methods with a :func:`set_dirty_and_flush_changes` decorator in addition to
that.
"""
kmm="_mutating_methods_"
ifkmminclsdict:
mutating_methods=clsdict[kmm]
forbaseinbases:
methods= (tfortininspect.getmembers(base,inspect.isroutine)ifnott[0].startswith("_"))
formethod_name,methodinmethods:
ifmethod_nameinclsdict:
continue
method_with_values=needs_values(method)
ifmethod_nameinmutating_methods:
method_with_values=set_dirty_and_flush_changes(method_with_values)
# END mutating methods handling
clsdict[method_name]=method_with_values
# END for each name/method pair
# END for each base
# END if mutating methods configuration is set
new_type=super().__new__(cls,name,bases,clsdict)
returnnew_type
defneeds_values(func:Callable[...,_T])->Callable[...,_T]:
"""Return a method for ensuring we read values (on demand) before we try to access
them."""
@wraps(func)
defassure_data_present(self:"GitConfigParser",*args:Any,**kwargs:Any)->_T:
self.read()
returnfunc(self,*args,**kwargs)
# END wrapper method
returnassure_data_present
defset_dirty_and_flush_changes(non_const_func:Callable[...,_T])->Callable[...,_T]:
"""Return a method that checks whether given non constant function may be called.
If so, the instance will be set dirty. Additionally, we flush the changes right to
disk.
"""
defflush_changes(self:"GitConfigParser",*args:Any,**kwargs:Any)->_T:
rval=non_const_func(self,*args,**kwargs)
self._dirty=True
self.write()
returnrval
# END wrapper method
flush_changes.__name__=non_const_func.__name__
returnflush_changes
classSectionConstraint(Generic[T_ConfigParser]):
"""Constrains a ConfigParser to only option commands which are constrained to
always use the section we have been initialized with.
It supports all ConfigParser methods that operate on an option.
:note:
If used as a context manager, will release the wrapped ConfigParser.
"""
__slots__= ("_config","_section_name")
_valid_attrs_= (
"get_value",
"set_value",
"get",
"set",
"getint",
"getfloat",
"getboolean",
"has_option",
"remove_section",
"remove_option",
"options",
)
def__init__(self,config:T_ConfigParser,section:str)->None:
self._config=config
self._section_name=section
def__del__(self)->None:
# Yes, for some reason, we have to call it explicitly for it to work in PY3 !
# Apparently __del__ doesn't get call anymore if refcount becomes 0
# Ridiculous ... .
self._config.release()
def__getattr__(self,attr:str)->Any:
ifattrinself._valid_attrs_:
returnlambda*args,**kwargs:self._call_config(attr,*args,**kwargs)
returnsuper().__getattribute__(attr)
def_call_config(self,method:str,*args:Any,**kwargs:Any)->Any:
"""Call the configuration at the given method which must take a section name as
first argument."""
returngetattr(self._config,method)(self._section_name,*args,**kwargs)
@property
defconfig(self)->T_ConfigParser:
"""return: ConfigParser instance we constrain"""
returnself._config
defrelease(self)->None:
"""Equivalent to :meth:`GitConfigParser.release`, which is called on our
underlying parser instance."""
returnself._config.release()
def__enter__(self)->"SectionConstraint[T_ConfigParser]":
self._config.__enter__()
returnself
def__exit__(self,exception_type:str,exception_value:str,traceback:str)->None:
self._config.__exit__(exception_type,exception_value,traceback)
class_OMD(OrderedDict_OMD):
"""Ordered multi-dict."""
def__setitem__(self,key:str,value:_T)->None:
super().__setitem__(key, [value])
defadd(self,key:str,value:Any)->None:
ifkeynotinself:
super().__setitem__(key, [value])
return
super().__getitem__(key).append(value)
defsetall(self,key:str,values:List[_T])->None:
super().__setitem__(key,values)
def__getitem__(self,key:str)->Any:
returnsuper().__getitem__(key)[-1]
defgetlast(self,key:str)->Any:
returnsuper().__getitem__(key)[-1]
defsetlast(self,key:str,value:Any)->None:
ifkeynotinself:
super().__setitem__(key, [value])
return
prior=super().__getitem__(key)
prior[-1]=value
defget(self,key:str,default:Union[_T,None]=None)->Union[_T,None]:
returnsuper().get(key, [default])[-1]
defgetall(self,key:str)->List[_T]:
returnsuper().__getitem__(key)
defitems(self)->List[Tuple[str,_T]]:# type: ignore[override]
"""List of (key, last value for key)."""
return [(k,self[k])forkinself]
defitems_all(self)->List[Tuple[str,List[_T]]]:
"""List of (key, list of values for key)."""
return [(k,self.getall(k))forkinself]
defget_config_path(config_level:Lit_config_levels)->str:
# We do not support an absolute path of the gitconfig on Windows.
# Use the global config instead.
ifsys.platform=="win32"andconfig_level=="system":
config_level="global"
ifconfig_level=="system":
return"/etc/gitconfig"
elifconfig_level=="user":
config_home=os.environ.get("XDG_CONFIG_HOME")orosp.join(os.environ.get("HOME","~"),".config")
returnosp.normpath(osp.expanduser(osp.join(config_home,"git","config")))
elifconfig_level=="global":
returnosp.normpath(osp.expanduser("~/.gitconfig"))
elifconfig_level=="repository":
raiseValueError("No repo to get repository configuration from. Use Repo._get_config_path")
else:
# Should not reach here. Will raise ValueError if does. Static typing will warn
# about missing elifs.
assert_never(# type: ignore[unreachable]
config_level,
ValueError(f"Invalid configuration level:{config_level!r}"),
)
classGitConfigParser(cp.RawConfigParser,metaclass=MetaParserBuilder):
"""Implements specifics required to read git style configuration files.
This variation behaves much like the :manpage:`git-config(1)` command, such that the
configuration will be read on demand based on the filepath given during
initialization.
The changes will automatically be written once the instance goes out of scope, but
can be triggered manually as well.
The configuration file will be locked if you intend to change values preventing
other instances to write concurrently.
:note:
The config is case-sensitive even when queried, hence section and option names
must match perfectly.
:note:
If used as a context manager, this will release the locked file.
"""
# { Configuration
t_lock=LockFile
"""The lock type determines the type of lock to use in new configuration readers.
They must be compatible to the :class:`~git.util.LockFile` interface.
A suitable alternative would be the :class:`~git.util.BlockingLockFile`.
"""
re_comment=re.compile(r"^\s*[#;]")
# } END configuration
optvalueonly_source=r"\s*(?P<option>[^:=\s][^:=]*)"
OPTVALUEONLY=re.compile(optvalueonly_source)
OPTCRE=re.compile(optvalueonly_source+r"\s*(?P<vi>[:=])\s*"+r"(?P<value>.*)$")
deloptvalueonly_source
_mutating_methods_= ("add_section","remove_section","remove_option","set")
"""Names of :class:`~configparser.RawConfigParser` methods able to change the
instance."""
def__init__(
self,
file_or_files:Union[None,PathLike,"BytesIO",Sequence[Union[PathLike,"BytesIO"]]]=None,
read_only:bool=True,
merge_includes:bool=True,
config_level:Union[Lit_config_levels,None]=None,
repo:Union["Repo",None]=None,
)->None:
"""Initialize a configuration reader to read the given `file_or_files` and to
possibly allow changes to it by setting `read_only` False.
:param file_or_files:
A file path or file object, or a sequence of possibly more than one of them.
:param read_only:
If ``True``, the ConfigParser may only read the data, but not change it.
If ``False``, only a single file path or file object may be given. We will
write back the changes when they happen, or when the ConfigParser is
released. This will not happen if other configuration files have been
included.
:param merge_includes:
If ``True``, we will read files mentioned in ``[include]`` sections and
merge their contents into ours. This makes it impossible to write back an
individual configuration file. Thus, if you want to modify a single
configuration file, turn this off to leave the original dataset unaltered
when reading it.
:param repo:
Reference to repository to use if ``[includeIf]`` sections are found in
configuration files.
"""
cp.RawConfigParser.__init__(self,dict_type=_OMD)
self._dict:Callable[...,_OMD]
self._defaults:_OMD
self._sections:_OMD
# Used in Python 3. Needs to stay in sync with sections for underlying
# implementation to work.
ifnothasattr(self,"_proxies"):
self._proxies=self._dict()
iffile_or_filesisnotNone:
self._file_or_files:Union[PathLike,"BytesIO",Sequence[Union[PathLike,"BytesIO"]]]=file_or_files
else:
ifconfig_levelisNone:
ifread_only:
self._file_or_files= [
get_config_path(cast(Lit_config_levels,f))forfinCONFIG_LEVELSiff!="repository"
]
else:
raiseValueError("No configuration level or configuration files specified")
else:
self._file_or_files= [get_config_path(config_level)]
self._read_only=read_only
self._dirty=False
self._is_initialized=False
self._merge_includes=merge_includes
self._repo=repo
self._lock:Union["LockFile",None]=None
self._acquire_lock()
def_acquire_lock(self)->None:
ifnotself._read_only:
ifnotself._lock:
ifisinstance(self._file_or_files, (str,os.PathLike)):
file_or_files=self._file_or_files
elifisinstance(self._file_or_files, (tuple,list,Sequence)):
raiseValueError(
"Write-ConfigParsers can operate on a single file only, multiple files have been passed"
)
else:
file_or_files=self._file_or_files.name
# END get filename from handle/stream
# Initialize lock base - we want to write.
self._lock=self.t_lock(file_or_files)
# END lock check
self._lock._obtain_lock()
# END read-only check
def__del__(self)->None:
"""Write pending changes if required and release locks."""
# NOTE: Only consistent in Python 2.
self.release()
def__enter__(self)->"GitConfigParser":
self._acquire_lock()
returnself
def__exit__(self,*args:Any)->None:
self.release()
defrelease(self)->None:
"""Flush changes and release the configuration write lock. This instance must
not be used anymore afterwards.
In Python 3, it's required to explicitly release locks and flush changes, as
``__del__`` is not called deterministically anymore.
"""
# Checking for the lock here makes sure we do not raise during write()
# in case an invalid parser was created who could not get a lock.
ifself.read_onlyor (self._lockandnotself._lock._has_lock()):
return
try:
self.write()
exceptIOError:
_logger.error("Exception during destruction of GitConfigParser",exc_info=True)
exceptReferenceError:
# This happens in Python 3... and usually means that some state cannot be
# written as the sections dict cannot be iterated. This usually happens when
# the interpreter is shutting down. Can it be fixed?
pass
finally:
ifself._lockisnotNone:
self._lock._release_lock()
defoptionxform(self,optionstr:str)->str:
"""Do not transform options in any way when writing."""
returnoptionstr
def_read(self,fp:Union[BufferedReader,IO[bytes]],fpname:str)->None:
"""Originally a direct copy of the Python 2.4 version of
:meth:`RawConfigParser._read <configparser.RawConfigParser._read>`, to ensure it
uses ordered dicts.
The ordering bug was fixed in Python 2.4, and dict itself keeps ordering since
Python 3.7. This has some other changes, especially that it ignores initial
whitespace, since git uses tabs. (Big comments are removed to be more compact.)
"""
cursect=None# None, or a dictionary.
optname=None
lineno=0
is_multi_line=False
e=None# None, or an exception.
defstring_decode(v:str)->str:
ifvandv.endswith("\\"):
v=v[:-1]
# END cut trailing escapes to prevent decode error
returnv.encode(defenc).decode("unicode_escape")
# END string_decode
whileTrue:
# We assume to read binary!
line=fp.readline().decode(defenc)
ifnotline:
break
lineno=lineno+1
# Comment or blank line?
ifline.strip()==""orself.re_comment.match(line):
continue
ifline.split(None,1)[0].lower()=="rem"andline[0]in"rR":
# No leading whitespace.
continue
# Is it a section header?
mo=self.SECTCRE.match(line.strip())
ifnotis_multi_lineandmo:
sectname:str=mo.group("header").strip()
ifsectnameinself._sections:
cursect=self._sections[sectname]
elifsectname==cp.DEFAULTSECT:
cursect=self._defaults
else:
cursect=self._dict((("__name__",sectname),))
self._sections[sectname]=cursect
self._proxies[sectname]=None
# So sections can't start with a continuation line.
optname=None
# No section header in the file?
elifcursectisNone:
raisecp.MissingSectionHeaderError(fpname,lineno,line)
# An option line?
elifnotis_multi_line:
mo=self.OPTCRE.match(line)
ifmo:
# We might just have handled the last line, which could contain a quotation we want to remove.
optname,vi,optval=mo.group("option","vi","value")
optname=self.optionxform(optname.rstrip())
ifviin ("=",":")and";"inoptvalandnotoptval.strip().startswith('"'):
pos=optval.find(";")
ifpos!=-1andoptval[pos-1].isspace():
optval=optval[:pos]
optval=optval.strip()
iflen(optval)<2oroptval[0]!='"':
# Does not open quoting.
pass
elifoptval[-1]!='"':
# Opens quoting and does not close: appears to start multi-line quoting.
is_multi_line=True
optval=string_decode(optval[1:])
elifoptval.find("\\",1,-1)==-1andoptval.find('"',1,-1)==-1:
# Opens and closes quoting. Single line, and all we need is quote removal.
optval=optval[1:-1]
# TODO: Handle other quoted content, especially well-formed backslash escapes.
# Preserves multiple values for duplicate optnames.
cursect.add(optname,optval)
else:
# Check if it's an option with no value - it's just ignored by git.
ifnotself.OPTVALUEONLY.match(line):
ifnote:
e=cp.ParsingError(fpname)
e.append(lineno,repr(line))
continue
else:
line=line.rstrip()
ifline.endswith('"'):
is_multi_line=False
line=line[:-1]
# END handle quotations
optval=cursect.getlast(optname)
cursect.setlast(optname,optval+string_decode(line))
# END parse section or option
# END while reading
# If any parsing errors occurred, raise an exception.
ife:
raisee
def_has_includes(self)->Union[bool,int]:
returnself._merge_includesandlen(self._included_paths())
def_included_paths(self)->List[Tuple[str,str]]:
"""List all paths that must be included to configuration.
:return:
The list of paths, where each path is a tuple of (option, value).
"""
def_all_items(section:str)->List[Tuple[str,str]]:
"""Return all (key, value) pairs for a section, including duplicate keys."""
return [
(key,value)
forkey,valuesinself._sections[section].items_all()
ifkey!="__name__"
forvalueinvalues
]
paths= []
forsectioninself.sections():
ifsection=="include":
paths+=_all_items(section)
match=CONDITIONAL_INCLUDE_REGEXP.search(section)
ifmatchisNoneorself._repoisNone:
continue
keyword=match.group(1)
value=match.group(2).strip()
ifkeywordin ["gitdir","gitdir/i"]:
value=osp.expanduser(value)
ifnotany(value.startswith(s)forsin ["./","/"]):
value="**/"+value
ifvalue.endswith("/"):
value+="**"
# Ensure that glob is always case insensitive if required.
ifkeyword.endswith("/i"):
value=re.sub(
r"[a-zA-Z]",
lambdam:f"[{m.group().lower()!r}{m.group().upper()!r}]",
value,
)
ifself._repo.git_dir:
iffnmatch.fnmatchcase(os.fspath(self._repo.git_dir),value):
paths+=_all_items(section)
elifkeyword=="onbranch":
try:
branch_name=self._repo.active_branch.name
exceptTypeError:
# Ignore section if active branch cannot be retrieved.
continue
iffnmatch.fnmatchcase(branch_name,value):
paths+=_all_items(section)
elifkeyword=="hasconfig:remote.*.url":
forremoteinself._repo.remotes:
iffnmatch.fnmatchcase(remote.url,value):
paths+=_all_items(section)
break
returnpaths
defread(self)->None:# type: ignore[override]
"""Read the data stored in the files we have been initialized with.
This will ignore files that cannot be read, possibly leaving an empty
configuration.
:raise IOError:
If a file cannot be handled.
"""
ifself._is_initialized:
return
self._is_initialized=True
files_to_read:List[Union[PathLike,IO]]= [""]
ifisinstance(self._file_or_files, (str,os.PathLike)):
# For str or Path, as str is a type of Sequence.
files_to_read= [self._file_or_files]
elifnotisinstance(self._file_or_files, (tuple,list,Sequence)):
# Could merge with above isinstance once runtime type known.
files_to_read= [self._file_or_files]
else:# For lists or tuples.
files_to_read=list(self._file_or_files)
# END ensure we have a copy of the paths to handle
seen=set(files_to_read)
num_read_include_files=0
whilefiles_to_read:
file_path=files_to_read.pop(0)
file_ok=False
ifhasattr(file_path,"seek"):
# Must be a file-object.
# TODO: Replace cast with assert to narrow type, once sure.
file_path=cast(IO[bytes],file_path)
self._read(file_path,file_path.name)
else:
try:
withopen(file_path,"rb")asfp:
file_ok=True
self._read(fp,fp.name)
exceptIOError:
continue
# Read includes and append those that we didn't handle yet. We expect all
# paths to be normalized and absolute (and will ensure that is the case).
ifself._has_includes():
for_,include_pathinself._included_paths():
ifinclude_path.startswith("~"):
include_path=osp.expanduser(include_path)
ifnotosp.isabs(include_path):
ifnotfile_ok:
continue
# END ignore relative paths if we don't know the configuration file path
file_path=cast(PathLike,file_path)
assertosp.isabs(file_path),"Need absolute paths to be sure our cycle checks will work"
include_path=osp.join(osp.dirname(file_path),include_path)
# END make include path absolute
include_path=osp.normpath(include_path)
ifinclude_pathinseenornotos.access(include_path,os.R_OK):
continue
seen.add(include_path)
# Insert included file to the top to be considered first.
files_to_read.insert(0,include_path)
num_read_include_files+=1
# END each include path in configuration file
# END handle includes
# END for each file object to read
# If there was no file included, we can safely write back (potentially) the
# configuration file without altering its meaning.
ifnum_read_include_files==0:
self._merge_includes=False
def_write(self,fp:IO)->None:
"""Write an .ini-format representation of the configuration state in
git compatible format."""
defwrite_section(name:str,section_dict:_OMD)->None:
fp.write(("[%s]\n"%name).encode(defenc))
values:Sequence[str]# Runtime only gets str in tests, but should be whatever _OMD stores.
v:str
forkey,valuesinsection_dict.items_all():
ifkey=="__name__":
continue
forvinvalues:
fp.write(("\t%s = %s\n"% (key,self._value_to_string(v).replace("\n","\n\t"))).encode(defenc))
# END if key is not __name__
# END section writing
ifself._defaults:
write_section(cp.DEFAULTSECT,self._defaults)
value:_OMD
forname,valueinself._sections.items():
write_section(name,value)
defitems(self,section_name:str)->List[Tuple[str,str]]:# type: ignore[override]
""":return: list((option, value), ...) pairs of all items in the given section"""
return [(k,v)fork,vinsuper().items(section_name)ifk!="__name__"]
defitems_all(self,section_name:str)->List[Tuple[str,List[str]]]:
""":return: list((option, [values...]), ...) pairs of all items in the given section"""
rv=_OMD(self._defaults)
fork,vsinself._sections[section_name].items_all():
ifk=="__name__":
continue
ifkinrvandrv.getall(k)==vs:
continue
forvinvs:
rv.add(k,v)
returnrv.items_all()
@needs_values
defwrite(self)->None:
"""Write changes to our file, if there are changes at all.
:raise IOError:
If this is a read-only writer instance or if we could not obtain a file
lock.
"""
self._assure_writable("write")
ifnotself._dirty:
return
ifisinstance(self._file_or_files, (list,tuple)):
raiseAssertionError(
"Cannot write back if there is not exactly a single file to write to, have %i files"
%len(self._file_or_files)
)
# END assert multiple files
ifself._has_includes():
_logger.debug(
"Skipping write-back of configuration file as include files were merged in."
+"Set merge_includes=False to prevent this."
)
return
# END stop if we have include files
fp=self._file_or_files
# We have a physical file on disk, so get a lock.
is_file_lock=isinstance(fp, (str,os.PathLike,IOBase))# TODO: Use PathLike (having dropped 3.5).
ifis_file_lockandself._lockisnotNone:# Else raise error?
self._lock._obtain_lock()
ifnothasattr(fp,"seek"):
fp=cast(PathLike,fp)
withopen(fp,"wb")asfp_open:
self._write(fp_open)
else:
fp=cast("BytesIO",fp)
fp.seek(0)
# Make sure we do not overwrite into an existing file.
ifhasattr(fp,"truncate"):
fp.truncate()
self._write(fp)
def_assure_writable(self,method_name:str)->None:
ifself.read_only:
raiseIOError("Cannot execute non-constant method %s.%s"% (self,method_name))
defadd_section(self,section:"cp._SectionName")->None:
"""Assures added options will stay in order."""
returnsuper().add_section(section)
@property
defread_only(self)->bool:
""":return: ``True`` if this instance may change the configuration file"""
returnself._read_only
# FIXME: Figure out if default or return type can really include bool.
defget_value(
self,
section:str,
option:str,
default:Union[int,float,str,bool,None]=None,
)->Union[int,float,str,bool]:
"""Get an option's value.
If multiple values are specified for this option in the section, the last one
specified is returned.
:param default:
If not ``None``, the given default value will be returned in case the option
did not exist.
:return:
A properly typed value, either int, float or string
:raise TypeError:
In case the value could not be understood.
Otherwise the exceptions known to the ConfigParser will be raised.
"""
try:
valuestr=self.get(section,option)
exceptException:
ifdefaultisnotNone:
returndefault
raise
returnself._string_to_value(valuestr)
defget_values(
self,
section:str,
option:str,
default:Union[int,float,str,bool,None]=None,
)->List[Union[int,float,str,bool]]:
"""Get an option's values.
If multiple values are specified for this option in the section, all are
returned.
:param default:
If not ``None``, a list containing the given default value will be returned
in case the option did not exist.
:return:
A list of properly typed values, either int, float or string
:raise TypeError:
In case the value could not be understood.
Otherwise the exceptions known to the ConfigParser will be raised.
"""
try:
self.sections()
lst=self._sections[section].getall(option)
exceptException:
ifdefaultisnotNone:
return [default]
raise
return [self._string_to_value(valuestr)forvaluestrinlst]
def_string_to_value(self,valuestr:str)->Union[int,float,str,bool]:
types= (int,float)
fornumtypeintypes:
try:
val=numtype(valuestr)
# truncated value ?
ifval!=float(valuestr):
continue
returnval
except (ValueError,TypeError):
continue
# END for each numeric type
# Try boolean values as git uses them.
vl=valuestr.lower()
ifvl=="false":
returnFalse
ifvl=="true":
returnTrue
ifnotisinstance(valuestr,str):
raiseTypeError(
"Invalid value type: only int, long, float and str are allowed",
valuestr,
)
returnvaluestr
def_value_to_string(self,value:Union[str,bytes,int,float,bool])->str:
ifisinstance(value, (int,float,bool)):
returnstr(value)
returnforce_text(value)
@needs_values
@set_dirty_and_flush_changes
defset_value(self,section:str,option:str,value:Union[str,bytes,int,float,bool])->"GitConfigParser":
"""Set the given option in section to the given value.
This will create the section if required, and will not throw as opposed to the
default ConfigParser ``set`` method.
:param section:
Name of the section in which the option resides or should reside.
:param option:
Name of the options whose value to set.
:param value:
Value to set the option to. It must be a string or convertible to a string.
:return:
This instance
"""
ifnotself.has_section(section):
self.add_section(section)
self.set(section,option,self._value_to_string(value))
returnself
@needs_values
@set_dirty_and_flush_changes
defadd_value(self,section:str,option:str,value:Union[str,bytes,int,float,bool])->"GitConfigParser":
"""Add a value for the given option in section.
This will create the section if required, and will not throw as opposed to the
default ConfigParser ``set`` method. The value becomes the new value of the
option as returned by :meth:`get_value`, and appends to the list of values
returned by :meth:`get_values`.
:param section:
Name of the section in which the option resides or should reside.
:param option:
Name of the option.
:param value:
Value to add to option. It must be a string or convertible to a string.
:return:
This instance
"""
ifnotself.has_section(section):
self.add_section(section)
self._sections[section].add(option,self._value_to_string(value))
returnself
defrename_section(self,section:str,new_name:str)->"GitConfigParser":
"""Rename the given section to `new_name`.
:raise ValueError:
If:
* `section` doesn't exist.
* A section with `new_name` does already exist.
:return:
This instance
"""
ifnotself.has_section(section):
raiseValueError("Source section '%s' doesn't exist"%section)
ifself.has_section(new_name):
raiseValueError("Destination section '%s' already exists"%new_name)
super().add_section(new_name)
new_section=self._sections[new_name]
fork,vsinself.items_all(section):
new_section.setall(k,vs)
# END for each value to copy
# This call writes back the changes, which is why we don't have the respective
# decorator.
self.remove_section(section)
returnself