Coverage for python/lsst/ctrl/bps/panda/cli/cmd/panda_auth_commands.py: 81%

16 statements  

« prev     ^ index     » next       coverage.py v6.5.0, created at 2022-11-08 22:36 -0800

1# This file is part of ctrl_bps. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (https://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <https://www.gnu.org/licenses/>. 

21"""Subcommand definitions for the PanDA auth commands. 

22""" 

23 

24__all__ = [ 

25 "status", 

26 "reset", 

27 "clean", 

28] 

29 

30 

31import click 

32from lsst.daf.butler.cli.utils import MWCommand 

33 

34from ...panda_auth_drivers import panda_auth_clean_driver, panda_auth_reset_driver, panda_auth_status_driver 

35 

36 

37class PandaAuthCommand(MWCommand): 

38 """Command subclass with panda-auth-command specific overrides.""" 

39 

40 extra_epilog = "See 'panda_auth --help' for more options." 

41 

42 

43@click.command(cls=PandaAuthCommand) 

44def status(*args, **kwargs): 

45 """Print informatino about auth token.""" 

46 panda_auth_status_driver(*args, **kwargs) 

47 

48 

49@click.command(cls=PandaAuthCommand) 

50def reset(*args, **kwargs): 

51 """Get new auth token.""" 

52 panda_auth_reset_driver(*args, **kwargs) 

53 

54 

55@click.command(cls=PandaAuthCommand) 

56def clean(*args, **kwargs): 

57 """Clean up token and token cache files.""" 

58 panda_auth_clean_driver(*args, **kwargs)