summaryrefslogtreecommitdiffstats
path: root/plugins/plugin_examples/dialogue.py
blob: 1ff007abc53a88a4d95248d6bb1196cafa5c0324 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# coding=utf-8
# First Aid Kit - diagnostic and repair tool for Linux
# Copyright (C) 2009 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
# Red Hat author: Miloslav Trmač <mitr@redhat.com>

from pyfirstaidkit.plugins import Plugin,Flow
from pyfirstaidkit.reporting import PLUGIN
from pyfirstaidkit.returns import *
from pyfirstaidkit.issue import SimpleIssue

class DialoguePlugin(Plugin):
    """This plugin demonstrates asking the user for information."""
    name = "DialoguePlugin"
    version = "0.0.1"
    author = "Miloslav Trmač"

    def __init__(self, *args, **kwargs):
        Plugin.__init__(self, *args, **kwargs)
        self._issue = SimpleIssue(self.name, self.description)

    def prepare(self):
        self._issue.set(reporting = self._reporting, origin = self,
                        level = PLUGIN)
        self._result=ReturnSuccess

    def backup(self):
        self._result=ReturnSuccess

    def restore(self):
        self._result=ReturnSuccess

    def diagnose(self):
        self._result=ReturnSuccess
        self._reporting.info("DialoguePlugin in diagnose task", origin = self,
                             level = PLUGIN)
        tea = self._reporting.choice_question_wait \
            ("Would you care for some tea?", ((True, "Yes"), (False, "No")),
             origin = self, level = PLUGIN)
        if tea:
            self._reporting.info("Here's your tea", origin = self,
                                 level = PLUGIN)
        else:
            self._reporting.info("No tea for you!", origin = self,
                                 level = PLUGIN)
        s = self._reporting.text_question_wait("Enter your name: ",
                                               origin = self, level = PLUGIN)
        self._reporting.info("Name: %s" % repr(s), origin = self,
                             level = PLUGIN)
        s = self._reporting.password_question_wait("Old fake password: ",
                                                   origin = self,
                                                   level = PLUGIN)
        self._reporting.info("Old password: %s" % repr(s), origin = self,
                             level = PLUGIN)
        s = self._reporting.password_question_wait("New fake password: ",
                                                   confirm = True,
                                                   origin = self,
                                                   level = PLUGIN)
        self._reporting.info("New password: %s" % repr(s), origin = self,
                             level = PLUGIN)
        s = self._reporting.filename_question_wait("File name: ", origin = self,
                                                   level = PLUGIN)
        self._reporting.info("File name: %s" % repr(s), origin = self,
                             level = PLUGIN)
        self._issue.set(checked = True, happened = False,
                        reporting = self._reporting, origin = self,
                        level = PLUGIN)

    def fix(self):
        self._result=ReturnSuccess

    def clean(self):
        self._result=ReturnSuccess

def get_plugin():
    return DialoguePlugin