[Python] .INI 파일에 설정 읽고 쓰기 - 2. 간편화 코드

1. 연구 모듈/Python 2020. 12. 18. 07:30

.INI 파일에 설정을 읽고 쓰는 것을 간편하게 하기 위한 boilerplate(부뚜막?) 코드



개요


다음 의식의 흐름에 의해 처리된다.

  1. 소스 상의 디폴트 설정값을 적용한다.
  2. .INI 파일이 존재할 경우 .INI 파일 상의 설정값을 적용한다.
  3. 최종 설정값을 .INI 파일에 저장한다.


소스


easyconfig.py
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
# -*- coding: utf-8 -*-
#!/usr/bin/python
 
import configparser
 
class EasyConfig:
 
    SECTION_DEFAULT = 'DEFAULT'
    configFileName = None
 
    # 설정 파일명을 세팅함
    def setFileName(self, fileName):
        self.configFileName = fileName
 
    # 파일로부터 설정을 세팅함
    def loadConfig(self, configParam):
        if not self.configFileName:
            return
         
        configFile = configparser.ConfigParser()
        configFile.read(self.configFileName)
         
        for key in configParam.keys():
            if key in configFile[self.SECTION_DEFAULT]:
                if type(configParam[key]) == int:
                    try:
                        val = int(configFile[self.SECTION_DEFAULT][key])
                        configParam[key] = val
                    except:
                        pass
                elif type(configParam[key]) == float:
                    try:
                        val = float(configFile[self.SECTION_DEFAULT][key])
                        configParam[key] = val
                    except:
                        pass
                elif type(configParam[key]) == bool:
                    if configFile[self.SECTION_DEFAULT][key] == 'True':
                        configParam[key] = True
                    else:
                        configParam[key] = False
                else:
                    configParam[key] = configFile[self.SECTION_DEFAULT][key]
 
    # 현재 설정을 파일로 저장함
    def saveConfig(self, configParam):
        if not self.configFileName:
            return
         
        configFile = configparser.ConfigParser()
         
        for key in configParam:
            val = None
            if type(configParam[key]) == int:
                val = "%d" % configParam[key]
            elif type(configParam[key]) == float:
                val = "%f" % configParam[key]
            elif type(configParam[key]) == bool:
                if configParam[key]:
                    val = 'True'
                else:
                    val = 'False'
            elif type(configParam[key]) == str:
                val = configParam[key]
 
            if val:
                configFile[self.SECTION_DEFAULT][key] = val
         
        with open(self.configFileName, 'w') as writeFile:
            configFile.write(writeFile)
 
    # 현재 설정 내용을 출력함
    def dumpConfig(self, configParam):
        for key in configParam:
            if type(configParam[key]) == int:
                print("%s = %d[int]" % (key, configParam[key]))
            elif type(configParam[key]) == float:
                print("%s = %f[float]" % (key, configParam[key]))
            elif type(configParam[key]) == bool:
                if configParam[key]:
                    print("%s = True[bool]" % key)
                else:
                    print("%s = False[bool]" % key)
            elif type(configParam[key]) == str:
                print("%s = %s[str]" % (key, configParam[key]))
 
    def process(self, configParam):
        self.loadConfig(configParam)
        self.saveConfig(configParam)
  • DEFAULT 섹션만을 사용한다. 그 외의 섹션을 사용하려면 섹션 존재 여부를 확인하는 추가적인 처리가 필요하다.


사용 방법


client.py
1
2
3
4
5
6
7
8
9
10
11
12
13
from easyconfig import *
 
# 설정 디폴트 값을 맵으로 정의한다
configDef = {}
configDef["A"] = "B"
configDef["C"] = "D"
configDef["E"] = 12
configDef["F"] = 3.14
configDef["G"] = True
 
easyConfig = EasyConfig()
easyConfig.setFileName('ec_config.ini') # 설정 파일명을 정의한다
easyConfig.process(configDef) # 설정 파일을 처리한다
  • int, float, bool, 문자열 타입만 지원한다. 설정 맵 상에 그 외의 타입이 존재할 경우 저장은 되지 않는다.


참고사항

<테스트 환경>
- OS : Windows 10
- Python 버전 : 3.7.5
저작자표시 비영리 동일조건

'1. 연구 모듈 > Python' 카테고리의 다른 글

[Python] 키보드 후킹 - 2. 간편화 코드  (0) 2020.12.23
[Python] 키보드 후킹 - 1. 예제 코드  (0) 2020.12.21
[Python] .INI 파일에 설정 읽고 쓰기 - 1. 예제 코드  (0) 2020.10.16
[Python] 인스턴스 메소드 vs. 정적 메소드 vs. 클래스 메소드 요약  (0) 2020.09.21
[Python] 클래스 변수 vs. 인스턴스 변수 요약  (0) 2020.08.31
,

티스토리툴바