[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
Re: [PATCH v5 30/36] qapi/gen.py: update write() to be more idiomatic
From: |
Markus Armbruster |
Subject: |
Re: [PATCH v5 30/36] qapi/gen.py: update write() to be more idiomatic |
Date: |
Wed, 07 Oct 2020 14:32:09 +0200 |
User-agent: |
Gnus/5.13 (Gnus v5.13) Emacs/27.1 (gnu/linux) |
John Snow <jsnow@redhat.com> writes:
> Make the file handling here just a tiny bit more idiomatic.
> (I realize this is heavily subjective.)
>
> Use exist_ok=True for os.makedirs and remove the exception,
> use fdopen() to wrap the file descriptor in a File-like object,
> and use a context manager for managing the file pointer.
>
> Signed-off-by: John Snow <jsnow@redhat.com>
> Reviewed-by: Eduardo Habkost <ehabkost@redhat.com>
> Reviewed-by: Cleber Rosa <crosa@redhat.com>
> ---
> scripts/qapi/gen.py | 25 +++++++++++--------------
> 1 file changed, 11 insertions(+), 14 deletions(-)
>
> diff --git a/scripts/qapi/gen.py b/scripts/qapi/gen.py
> index 3624162bb77..579ee283297 100644
> --- a/scripts/qapi/gen.py
> +++ b/scripts/qapi/gen.py
> @@ -14,7 +14,6 @@
> # See the COPYING file in the top-level directory.
>
> from contextlib import contextmanager
> -import errno
> import os
> import re
> from typing import (
> @@ -67,21 +66,19 @@ def write(self, output_dir: str) -> None:
> return
> pathname = os.path.join(output_dir, self.fname)
> odir = os.path.dirname(pathname)
> +
> if odir:
> - try:
> - os.makedirs(odir)
> - except os.error as e:
> - if e.errno != errno.EEXIST:
> - raise
> + os.makedirs(odir, exist_ok=True)
I wouldn't call this part "heavily subjective". When wrote the old
code, exist_ok was still off limits (it's new in 3.2).
> +
> + # use os.open for O_CREAT to create and read a non-existant file
> fd = os.open(pathname, os.O_RDWR | os.O_CREAT, 0o666)
> - f = open(fd, 'r+', encoding='utf-8')
> - text = self.get_content()
> - oldtext = f.read(len(text) + 1)
> - if text != oldtext:
> - f.seek(0)
> - f.truncate(0)
> - f.write(text)
> - f.close()
> + with os.fdopen(fd, 'r+', encoding='utf-8') as fp:
> + text = self.get_content()
> + oldtext = fp.read(len(text) + 1)
> + if text != oldtext:
> + fp.seek(0)
> + fp.truncate(0)
> + fp.write(text)
>
>
> def _wrap_ifcond(ifcond: List[str], before: str, after: str) -> str:
Reviewed-by: Markus Armbruster <armbru@redhat.com>
- Re: [PATCH v5 27/36] qapi/gen.py: add type hint annotations, (continued)
[PATCH v5 31/36] qapi/gen.py: delint with pylint, John Snow, 2020/10/05
[PATCH v5 29/36] qapi/gen.py: Remove unused parameter, John Snow, 2020/10/05
[PATCH v5 30/36] qapi/gen.py: update write() to be more idiomatic, John Snow, 2020/10/05
- Re: [PATCH v5 30/36] qapi/gen.py: update write() to be more idiomatic,
Markus Armbruster <=
[PATCH v5 32/36] qapi/types.py: add type hint annotations, John Snow, 2020/10/05
[PATCH v5 34/36] qapi/visit.py: assert tag_member contains a QAPISchemaEnumType, John Snow, 2020/10/05
[PATCH v5 33/36] qapi/types.py: remove one-letter variables, John Snow, 2020/10/05