drath/libratbag
1
0
Fork 1
mirror of https://github.com/libratbag/libratbag.git synced 2026-11-07 23:23:52 +01:00
Code Issues Projects Releases Wiki Activity

tools: use with-statements to open files

Browse Source 
I'll be honest, writing to files with `print` looks weird...
This commit is contained in:
Yaroslav Chvanov
2023-02-25 00:53:24 +03:00
parent 74c2cb686e
commit e49c18b909
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/merge_ratbagd.py
View File

@@ -22,6 +22,7 @@
# DEALINGS IN THE SOFTWARE.
import argparse
import contextlib
import os
import stat
import sys
@@ -51,15 +52,11 @@ def main(argv):
parser.add_argument("--version", action="store", default="git_master")
ns = parser.parse_args(sys.argv[1:])
if ns.output:
ns.output_file = open(ns.output, 'w', encoding='utf-8')
st = os.stat(ns.output)
os.chmod(ns.output, st.st_mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
sys.stdout = ns.output_file
print_ratbagctl(ns.ratbagctl, ns.ratbagd, ns.version)
try:
ns.output_file.close()
except AttributeError:
pass
with open(ns.output, 'w', encoding='utf-8') as output_file:
st = os.stat(ns.output)
os.chmod(ns.output, st.st_mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
with contextlib.redirect_stdout(output_file):
print_ratbagctl(ns.ratbagctl, ns.ratbagd, ns.version)
if __name__ == "__main__":