deleteProfile.py

Go to the documentation of this file.

#!/usr/bin/python3
 
# smolt - Fedora hardware profiler
#
# Copyright (C) 2007 Mike McGrath
# Copyright (C) 2009 Sebastian Pipping <sebastian@pipping.org>
#
# 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
 
from builtins import str
import sys
import requests
from optparse import OptionParser
 
 
def main():
    sys.path.append('/usr/share/smolt/client')
 
    from i18n import _
    import smolt
    from smolt import error, debug, get_profile_link, PubUUIDError
    from uuiddb import create_default_uuiddb
 
    def serverMessage(page):
        for line in page.split("\n"):
            if 'ServerMessage:' in line:
                error(_('Server Message: "%s"' % line.split('ServerMessage: ')[1]))
                if 'Critical' in line:
                    sys.exit(3)
 
    parser = OptionParser(version = smolt.smoltProtocol)
 
    parser.add_option('-d', '--debug',
                    dest = 'DEBUG',
                    default = False,
                    action = 'store_true',
                    help = _('enable debug information'))
    parser.add_option('-s', '--server',
                    dest = 'smoonURL',
                    default = smolt.smoonURL,
                    metavar = 'smoonURL',
                    help = _('specify the URL of the server (default "%default")'))
    parser.add_option('-p', '--printOnly',
                    dest = 'printOnly',
                    default = False,
                    action = 'store_true',
                    help = _('print information only, do not send'))
    parser.add_option('-u', '--useragent',
                    dest = 'user_agent',
                    default = smolt.user_agent,
                    metavar = 'USERAGENT',
                    help = _('specify HTTP user agent (default "%default")'))
    parser.add_option('-t', '--timeout',
                    dest = 'timeout',
                    type = 'float',
                    default = smolt.timeout,
                    help = _('specify HTTP timeout in seconds (default %default seconds)'))
    parser.add_option('--uuidFile',
                    dest = 'uuidFile',
                    default = smolt.hw_uuid_file,
                    help = _('specify which uuid to use, useful for debugging and testing mostly.'))
 
 
    (opts, args) = parser.parse_args()
 
    smolt.DEBUG = opts.DEBUG
    smolt.hw_uuid_file = opts.uuidFile
 
    uuid = smolt.read_uuid()
    delHostString = 'uuid=%s' % uuid
 
    session = requests.Session()
    session.headers.update({'USER-AGENT': opts.user_agent})
    session.headers.update({'Content-type': 'application/x-www-form-urlencoded'})
    session.headers.update({'Content-length': '%i' % len(delHostString)})
 
    # Try retrieving current pub_uuid  (from cache or remotely if necessary)
    pub_uuid = None
    try:
        pub_uuid = smolt.read_pub_uuid(create_default_uuiddb(), uuid, silent=True)
    except PubUUIDError:
        pass
 
    exceptions = (requests.exceptions.HTTPError,
                  requests.exceptions.URLRequired,
                  requests.exceptions.Timeout,
                  requests.exceptions.ConnectionError,
                  requests.exceptions.InvalidURL)
 
    try:
        o = session.post(opts.smoonURL + 'client/delete', data=delHostString,
                         timeout=opts.timeout)
    except exceptions as e:
        sys.stderr.write(_('Error contacting Server:'))
        sys.stderr.write(str(e))
        sys.stderr.write('\n')
        sys.exit(1)
    else:
        serverMessage(o.text)
        session.close()
 
    if pub_uuid is None:
        profile_url = opts.smoonURL + 'client/show?%s' % delHostString
    else:
        profile_url = get_profile_link(opts.smoonURL, pub_uuid)
    print((_('Profile removed, please verify at'), profile_url))
 
 
if __name__ == '__main__':
    main()