pytorch-lightning

1
# Copyright The Lightning AI team.
2
#
3
# Licensed under the Apache License, Version 2.0 (the "License");
4
# you may not use this file except in compliance with the License.
5
# You may obtain a copy of the License at
6
#
7
#     http://www.apache.org/licenses/LICENSE-2.0
8
#
9
# Unless required by applicable law or agreed to in writing, software
10
# distributed under the License is distributed on an "AS IS" BASIS,
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14

15
from typing import Optional
16

17
from lightning.app.core.constants import REDIS_HOST, REDIS_PASSWORD, REDIS_PORT
18
from lightning.app.utilities.imports import _is_redis_available
19

20

21
def check_if_redis_running(
22
    host: Optional[str] = "", port: Optional[int] = 6379, password: Optional[str] = None
23
) -> bool:
24
    if not _is_redis_available():
25
        return False
26
    import redis
27

28
    try:
29
        host = host or REDIS_HOST
30
        port = port or REDIS_PORT
31
        password = password or REDIS_PASSWORD
32
        return redis.Redis(host=host, port=port, password=password).ping()
33
    except redis.exceptions.ConnectionError:
34
        return False
35