Skip to content

Commit ac77ef7

Browse files
author
v.shepard
committed
PBCKP-152 use black for formatting
1 parent f9b6bdb commit ac77ef7

File tree

8 files changed

+127
-110
lines changed

8 files changed

+127
-110
lines changed

‎testgres/cache.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44

55
from six import raise_from
66

7-
from .op_ops.local_ops import LocalOperations
8-
from .op_ops.os_ops import OsOperations
7+
from .os_ops.local_ops import LocalOperations
8+
from .os_ops.os_ops import OsOperations
99
from .config import testgres_config
1010

1111
from .consts import XLOG_CONTROL_FILE
@@ -25,6 +25,7 @@ def cached_initdb(data_dir, logfile=None, params=None, os_ops: OsOperations = Lo
2525
"""
2626
Perform initdb or use cached node files.
2727
"""
28+
testgres_config.os_ops = os_ops
2829

2930
def call_initdb(initdb_dir, log=None):
3031
try:
@@ -60,7 +61,7 @@ def call_initdb(initdb_dir, log=None):
6061

6162
# XXX: build new WAL segment with our system id
6263
_params = [get_bin_path("pg_resetwal"), "-D", data_dir, "-f"]
63-
execute_utility(_params, logfile, os_ops)
64+
execute_utility(_params, logfile, os_ops=os_ops)
6465

6566
except ExecUtilException as e:
6667
msg = "Failed to reset WAL for system id"

‎testgres/config.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,8 @@
55
import tempfile
66

77
from contextlib import contextmanager
8-
from shutil import rmtree
98

10-
from .op_ops.local_ops import LocalOperations
9+
from .os_ops.local_ops import LocalOperations
1110
from .consts import TMP_CACHE
1211

1312

@@ -44,7 +43,7 @@ class GlobalConfig(object):
4443
_cached_initdb_dir = None
4544
""" underlying class attribute for cached_initdb_dir property """
4645

47-
os_ops = None
46+
os_ops = LocalOperations()
4847
""" OsOperation object that allows work on remote host """
4948
@property
5049
def cached_initdb_dir(self):
@@ -137,9 +136,9 @@ def copy(self):
137136

138137

139138
@atexit.register
140-
def _rm_cached_initdb_dirs(os_ops=LocalOperations()):
139+
def _rm_cached_initdb_dirs():
141140
for d in cached_initdb_dirs:
142-
os_ops.rmdirs(d, ignore_errors=True)
141+
testgres_config.os_ops.rmdirs(d, ignore_errors=True)
143142

144143

145144
def push_config(**options):

‎testgres/defaults.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
import datetime
2-
import os
32
import struct
43
import uuid
54

6-
from .op_ops.local_ops import LocalOperations
5+
from .os_ops.local_ops import LocalOperations
76

87

98
def default_dbname():

‎testgres/node.py

Lines changed: 18 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,16 @@
11
# coding: utf-8
22

3-
import io
43
import os
54
import random
6-
import shutil
75
import signal
86
import threading
97
from queue import Queue
108

119
import psutil
12-
import subprocess
1310
import time
1411

15-
from op_ops.local_ops import LocalOperations
16-
from op_ops.os_ops import OsOperations
17-
from op_ops.remote_ops import RemoteOperations
12+
from .os_ops.local_ops import LocalOperations
13+
from .os_ops.remote_ops import RemoteOperations
1814

1915
try:
2016
from collections.abc import Iterable
@@ -32,7 +28,6 @@
3228

3329
from shutil import rmtree
3430
from six import raise_from, iteritems, text_type
35-
from tempfile import mkstemp, mkdtemp
3631

3732
from .enums import \
3833
NodeStatus, \
@@ -96,7 +91,6 @@
9691
eprint, \
9792
get_bin_path, \
9893
get_pg_version, \
99-
file_tail, \
10094
reserve_port, \
10195
release_port, \
10296
execute_utility, \
@@ -163,6 +157,7 @@ def __init__(self, name=None, port=None, base_dir=None,
163157
else:
164158
self.os_ops = RemoteOperations(host, hostname, ssh_key)
165159

160+
testgres_config.os_ops = self.os_ops
166161
# defaults for __exit__()
167162
self.cleanup_on_good_exit = testgres_config.node_cleanup_on_good_exit
168163
self.cleanup_on_bad_exit = testgres_config.node_cleanup_on_bad_exit
@@ -289,7 +284,7 @@ def base_dir(self):
289284
self._base_dir = self.os_ops.mkdtemp(prefix=TMP_NODE)
290285

291286
# NOTE: it's safe to create a new dir
292-
if not self.os_ops.exists(self._base_dir):
287+
if not self.os_ops.path_exists(self._base_dir):
293288
self.os_ops.makedirs(self._base_dir)
294289

295290
return self._base_dir
@@ -299,7 +294,7 @@ def logs_dir(self):
299294
path = os.path.join(self.base_dir, LOGS_DIR)
300295

301296
# NOTE: it's safe to create a new dir
302-
if not self.os_ops.exists(path):
297+
if not self.os_ops.path_exists(path):
303298
self.os_ops.makedirs(path)
304299

305300
return path
@@ -628,7 +623,7 @@ def status(self):
628623
"-D", self.data_dir,
629624
"status"
630625
] # yapf: disable
631-
execute_utility(_params, self.utils_log_file)
626+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
632627
return NodeStatus.Running
633628

634629
except ExecUtilException as e:
@@ -650,7 +645,7 @@ def get_control_data(self):
650645
_params += ["-D"] if self._pg_version >= PgVer('9.5') else []
651646
_params += [self.data_dir]
652647

653-
data = execute_utility(_params, self.utils_log_file)
648+
data = execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
654649

655650
out_dict = {}
656651

@@ -713,7 +708,7 @@ def start(self, params=[], wait=True):
713708
] + params # yapf: disable
714709

715710
try:
716-
execute_utility(_params, self.utils_log_file)
711+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
717712
except ExecUtilException as e:
718713
msg = 'Cannot start node'
719714
files = self._collect_special_files()
@@ -744,7 +739,7 @@ def stop(self, params=[], wait=True):
744739
"stop"
745740
] + params # yapf: disable
746741

747-
execute_utility(_params, self.utils_log_file)
742+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
748743

749744
self._maybe_stop_logger()
750745
self.is_started = False
@@ -786,7 +781,7 @@ def restart(self, params=[]):
786781
] + params # yapf: disable
787782

788783
try:
789-
execute_utility(_params, self.utils_log_file)
784+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
790785
except ExecUtilException as e:
791786
msg = 'Cannot restart node'
792787
files = self._collect_special_files()
@@ -813,7 +808,7 @@ def reload(self, params=[]):
813808
"reload"
814809
] + params # yapf: disable
815810

816-
execute_utility(_params, self.utils_log_file)
811+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
817812

818813
return self
819814

@@ -835,7 +830,7 @@ def promote(self, dbname=None, username=None):
835830
"promote"
836831
] # yapf: disable
837832

838-
execute_utility(_params, self.utils_log_file)
833+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
839834

840835
# for versions below 10 `promote` is asynchronous so we need to wait
841836
# until it actually becomes writable
@@ -870,7 +865,7 @@ def pg_ctl(self, params):
870865
"-w" # wait
871866
] + params # yapf: disable
872867

873-
return execute_utility(_params, self.utils_log_file)
868+
return execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
874869

875870
def free_port(self):
876871
"""
@@ -1035,10 +1030,9 @@ def dump(self,
10351030
# Generate tmpfile or tmpdir
10361031
def tmpfile():
10371032
if format == DumpFormat.Directory:
1038-
fname = mkdtemp(prefix=TMP_DUMP)
1033+
fname = self.os_ops.mkdtemp(prefix=TMP_DUMP)
10391034
else:
1040-
fd, fname = mkstemp(prefix=TMP_DUMP)
1041-
os.close(fd)
1035+
fname = self.os_ops.mkstemp(prefix=TMP_DUMP)
10421036
return fname
10431037

10441038
# Set default arguments
@@ -1056,7 +1050,7 @@ def tmpfile():
10561050
"-F", format.value
10571051
] # yapf: disable
10581052

1059-
execute_utility(_params, self.utils_log_file)
1053+
execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
10601054

10611055
return filename
10621056

@@ -1085,7 +1079,7 @@ def restore(self, filename, dbname=None, username=None):
10851079

10861080
# try pg_restore if dump is binary formate, and psql if not
10871081
try:
1088-
execute_utility(_params, self.utils_log_name)
1082+
execute_utility(_params, self.utils_log_name, os_ops=self.os_ops)
10891083
except ExecUtilException:
10901084
self.psql(filename=filename, dbname=dbname, username=username)
10911085

@@ -1417,7 +1411,7 @@ def pgbench_run(self, dbname=None, username=None, options=[], **kwargs):
14171411
# should be the last one
14181412
_params.append(dbname)
14191413

1420-
return execute_utility(_params, self.utils_log_file)
1414+
return execute_utility(_params, self.utils_log_file, os_ops=self.os_ops)
14211415

14221416
def connect(self,
14231417
dbname=None,

0 commit comments

Comments
 (0)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy