Coverage for osmclient/client.py: 22%

1# Copyright 2017-2018 Sandvine 

2# Copyright 2018 Telefonica 

3# 

4# All Rights Reserved. 

5# 

6# Licensed under the Apache License, Version 2.0 (the "License"); you may 

7# not use this file except in compliance with the License. You may obtain 

8# a copy of the License at 

9# 

10# http://www.apache.org/licenses/LICENSE-2.0 

11# 

12# Unless required by applicable law or agreed to in writing, software 

13# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

14# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

15# License for the specific language governing permissions and limitations 

16# under the License. 

17 

18""" 

19OSM client entry point 

20""" 

21 

22from osmclient.sol005 import client as sol005client 

23import logging 

24import verboselogs 

25 

26# pylint: disable=E1101 

27verboselogs.install() 

28 

29 

30def Client(version=1, host=None, *args, **kwargs): 

31 log_format_simple = "%(levelname)s %(message)s" 

32 log_format_complete = "%(asctime)s %(levelname)s %(name)s %(filename)s:%(lineno)s %(funcName)s(): %(message)s" 

33 log_formatter_simple = logging.Formatter( 

34 log_format_simple, datefmt="%Y-%m-%dT%H:%M:%S" 

35 ) 

36 handler = logging.StreamHandler() 

37 handler.setFormatter(log_formatter_simple) 

38 logger = logging.getLogger("osmclient") 

39 logger.setLevel(level=logging.WARNING) 

40 logger.addHandler(handler) 

41 verbose = kwargs.get("verbose", 0) 

42 if verbose > 0: 

43 log_formatter = logging.Formatter( 

44 log_format_complete, datefmt="%Y-%m-%dT%H:%M:%S" 

45 ) 

46 # handler = logging.StreamHandler() 

47 handler.setFormatter(log_formatter) 

48 # logger.addHandler(handler) 

49 if verbose == 1: 

50 logger.setLevel(level=logging.INFO) 

51 elif verbose == 2: 

52 logger.setLevel(level=logging.VERBOSE) 

53 elif verbose > 2: 

54 logger.setLevel(level=logging.DEBUG) 

55 if version == 1: 

56 return sol005client.Client(host, *args, **kwargs) 

57 else: 

58 raise Exception("Unsupported client version")