Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
SeleniumHQ
GitHub Repository: SeleniumHQ/Selenium
Path: blob/trunk/py/selenium/webdriver/remote/file_detector.py
1864 views
1
# Licensed to the Software Freedom Conservancy (SFC) under one
2
# or more contributor license agreements. See the NOTICE file
3
# distributed with this work for additional information
4
# regarding copyright ownership. The SFC licenses this file
5
# to you under the Apache License, Version 2.0 (the
6
# "License"); you may not use this file except in compliance
7
# with the License. You may obtain a copy of the License at
8
#
9
# http://www.apache.org/licenses/LICENSE-2.0
10
#
11
# Unless required by applicable law or agreed to in writing,
12
# software distributed under the License is distributed on an
13
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14
# KIND, either express or implied. See the License for the
15
# specific language governing permissions and limitations
16
# under the License.
17
18
from abc import ABCMeta, abstractmethod
19
from contextlib import suppress
20
from pathlib import Path
21
from typing import Optional
22
23
from selenium.types import AnyKey
24
from selenium.webdriver.common.utils import keys_to_typing
25
26
27
class FileDetector(metaclass=ABCMeta):
28
"""Used for identifying whether a sequence of chars represents the path to
29
a file."""
30
31
@abstractmethod
32
def is_local_file(self, *keys: AnyKey) -> Optional[str]:
33
raise NotImplementedError
34
35
36
class UselessFileDetector(FileDetector):
37
"""A file detector that never finds anything."""
38
39
def is_local_file(self, *keys: AnyKey) -> Optional[str]:
40
return None
41
42
43
class LocalFileDetector(FileDetector):
44
"""Detects files on the local disk."""
45
46
def is_local_file(self, *keys: AnyKey) -> Optional[str]:
47
file_path = "".join(keys_to_typing(keys))
48
49
with suppress(OSError):
50
if Path(file_path).is_file():
51
return file_path
52
return None
53
54