Skip to content

Commit 78d8607

Browse files
author
avandras
committed
Address some flake8 complaints
1 parent ec9b6c3 commit 78d8607

File tree

3 files changed

+20
-19
lines changed

3 files changed

+20
-19
lines changed

patroni/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -677,7 +677,7 @@ def do_GET_metrics(self) -> None:
677677
metrics.append("patroni_multisite_switches{0} {1}"
678678
.format(labels, patroni.multisite.site_switches))
679679

680-
self.write_response(200, '\n'.join(metrics)+'\n', content_type='text/plain')
680+
self.write_response(200, '\n'.join(metrics) + '\n', content_type='text/plain')
681681

682682
def do_GET_multisite(self):
683683
self._write_json_response(200, self.server.patroni.multisite.status())

patroni/ctl.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1385,8 +1385,8 @@ def _do_failover_or_switchover(action: str, cluster_name: str, group: Optional[i
13851385

13861386

13871387
def _do_site_switchover(cluster_name: str, group: Optional[int],
1388-
switchover_leader: Optional[str], candidate: Optional[str],
1389-
force: bool, scheduled: Optional[str] = None) -> None:
1388+
switchover_leader: Optional[str], candidate: Optional[str],
1389+
force: bool, scheduled: Optional[str] = None) -> None:
13901390
"""Perform a site switchover operation in the cluster.
13911391
13921392
Informational messages are printed in the console during the operation, as well as the list of members before and
@@ -1436,7 +1436,7 @@ def _do_site_switchover(cluster_name: str, group: Optional[int],
14361436

14371437
if cluster.leader and cluster.leader.multisite:
14381438
leader_site = (cluster.leader.multisite.get('name') if not cluster.leader.multisite.get('standby_config') else
1439-
cluster.leader.multisite.get('standby_config', {}).get('leader_site'))
1439+
cluster.leader.multisite.get('standby_config', {}).get('leader_site'))
14401440
else:
14411441
raise PatroniCtlException('Multisite is not active or there is no leader site, cannot switch sites')
14421442

@@ -1454,7 +1454,7 @@ def _do_site_switchover(cluster_name: str, group: Optional[int],
14541454
# multisite_cluster = multisite_dcs.get_cluster()
14551455

14561456
candidate_names = [str(m.multisite['name']) for m in cluster.members
1457-
if m.multisite and m.multisite['name'] != leader_site]
1457+
if m.multisite and m.multisite['name'] != leader_site]
14581458
# We sort the names for consistent output to the client
14591459
candidate_names.sort()
14601460

@@ -1478,7 +1478,7 @@ def _do_site_switchover(cluster_name: str, group: Optional[int],
14781478
if scheduled is None and not force:
14791479
next_hour = (datetime.datetime.now() + datetime.timedelta(hours=1)).strftime('%Y-%m-%dT%H:%M')
14801480
scheduled = click.prompt('When should the switchover take place (e.g. ' + next_hour + ' ) ',
1481-
type=str, default='now')
1481+
type=str, default='now')
14821482

14831483
scheduled_at = parse_scheduled(scheduled)
14841484
if scheduled_at:
@@ -1583,13 +1583,14 @@ def switchover(cluster_name: str, group: Optional[int], leader: Optional[str],
15831583
@ctl.command('site-switchover', help='Switchover to another data centre')
15841584
@arg_cluster_name
15851585
@option_citus_group
1586-
@click.option('--leader-site', '--primary-site', 'leader_site', help='The name of the current leader site', default=None)
1586+
@click.option('--leader-site', '--primary-site', 'leader_site', help='The name of the current leader site',
1587+
default=None)
15871588
@click.option('--candidate-site', 'candidate_site', help='The name of the candidate', default=None)
15881589
@click.option('--scheduled', help='Timestamp of a scheduled switchover in unambiguous format (e.g. ISO 8601)',
15891590
default=None)
15901591
@option_force
15911592
def site_switchover(cluster_name: str, group: Optional[int], leader_site: Optional[str],
1592-
candidate_site: Optional[str], force: bool, scheduled: Optional[str]) -> None:
1593+
candidate_site: Optional[str], force: bool, scheduled: Optional[str]) -> None:
15931594
"""Process ``multisite-switchover`` command of ``patronictl`` utility.
15941595
15951596
Perform a site switchover operation in the multisite cluster.
@@ -1609,7 +1610,6 @@ def site_switchover(cluster_name: str, group: Optional[int], leader_site: Option
16091610
_do_site_switchover(cluster_name, group, leader_site, candidate_site, force, scheduled)
16101611

16111612

1612-
16131613
def generate_topology(level: int, member: Dict[str, Any],
16141614
topology: Dict[Optional[str], List[Dict[str, Any]]]) -> Iterator[Dict[str, Any]]:
16151615
"""Recursively yield members with their names adjusted according to their *level* in the cluster topology.
@@ -1698,8 +1698,6 @@ def get_cluster_service_info(cluster: Dict[str, Any]) -> List[str]:
16981698
"""
16991699
service_info: List[str] = []
17001700

1701-
1702-
17031701
if 'multisite' in cluster:
17041702
info = f"Multisite {cluster['multisite'].get('name') or ''} is {cluster['multisite']['status'].lower()}"
17051703
standby_config = cluster['multisite'].get('standby_config', {})

patroni/multisite.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@
1212
import kubernetes
1313

1414
from .dcs import Cluster, Member
15-
from .dcs.kubernetes import catch_kubernetes_errors, Kubernetes
15+
from .dcs.kubernetes import catch_kubernetes_errors
1616
from .exceptions import DCSError
1717

1818
logger = logging.getLogger(__name__)
1919

20+
2021
@six.add_metaclass(abc.ABCMeta)
2122
class AbstractSiteController(object):
2223
# Set whether we are relying on this controller for providing standby config
@@ -60,11 +61,13 @@ def should_failover(self) -> bool:
6061
def on_shutdown(self, checkpoint_location):
6162
pass
6263

64+
6365
class SingleSiteController(AbstractSiteController):
6466
"""Do nothing controller for single site operation."""
6567
def status(self):
6668
return {"status": "Leader", "active": False}
6769

70+
6871
class MultisiteController(Thread, AbstractSiteController):
6972
is_active = True
7073

@@ -81,7 +84,7 @@ def __init__(self, config, on_change=None):
8184
if msconfig.get('update_crd'):
8285
self._state_updater = KubernetesStateManagement(msconfig.get('update_crd'),
8386
msconfig.get('crd_uid'),
84-
reporter=self.name, # Use pod name?
87+
reporter=self.name, # Use pod name?
8588
crd_api=msconfig.get('crd_api', 'acid.zalan.do/v1'))
8689
else:
8790
self._state_updater = None
@@ -200,7 +203,6 @@ def _check_transition(self, leader, note=None):
200203
self._state_updater.state_transition('Leader' if leader else 'Standby', note)
201204
self._status = leader
202205

203-
204206
def _resolve_multisite_leader(self):
205207
logger.info("Running multisite consensus.")
206208
try:
@@ -291,7 +293,7 @@ def _resolve_multisite_leader(self):
291293
try:
292294
self._update_history(cluster)
293295
self.touch_member()
294-
except DCSError as e:
296+
except DCSError:
295297
pass
296298

297299
def _observe_leader(self):
@@ -385,7 +387,7 @@ def __init__(self, crd_name, crd_uid, reporter, crd_api):
385387
self.crd_api_group, self.crd_api_version = crd_api.rsplit('/', 1)
386388

387389
# TODO: handle config loading when main DCS is not Kubernetes based
388-
#apiclient = k8s_client.ApiClient(False)
390+
# apiclient = k8s_client.ApiClient(False)
389391
kubernetes.config.load_incluster_config()
390392
apiclient = kubernetes.client.ApiClient()
391393
self._customobj_api = kubernetes.client.CustomObjectsApi(apiclient)
@@ -432,9 +434,10 @@ def store_updates(self):
432434

433435
@catch_kubernetes_errors
434436
def update_crd_state(self, update):
435-
self._customobj_api.patch_namespaced_custom_object_status(self.crd_api_group, self.crd_api_version, self.crd_namespace,
436-
'postgresqls', self.crd_name + '/status', update,
437-
field_manager='patroni')
437+
self._customobj_api.patch_namespaced_custom_object_status(self.crd_api_group, self.crd_api_version,
438+
self.crd_namespace,
439+
'postgresqls', self.crd_name + '/status', update,
440+
field_manager='patroni')
438441

439442
return True
440443

0 commit comments

Comments
 (0)