%PDF-
%PDF-
Mini Shell
Mini Shell
# coding=utf-8
# generate_completion_cache.py - generate cache for dnf bash completion
# Copyright © 2013 Elad Alfassa <elad@fedoraproject.org>
# Copyright (C) 2014-2015 Igor Gnatenko <i.gnatenko.brain@gmail.com>
# Copyright (C) 2015 Red Hat, Inc.
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# the GNU General Public License v.2, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY expressed or implied, including the implied warranties 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 Street, Fifth Floor, Boston, MA
# 02110-1301, USA.
from __future__ import absolute_import
from __future__ import unicode_literals
from dnf.i18n import ucd
from dnfpluginscore import logger
import dnf
import os.path
import sqlite3
class BashCompletionCache(dnf.Plugin):
name = 'generate_completion_cache'
def __init__(self, base, cli):
super(BashCompletionCache, self).__init__(base, cli)
self.base = base
self.cache_file = "/var/cache/dnf/packages.db"
@staticmethod
def _out(msg):
logger.debug('Completion plugin: %s', msg)
def sack(self):
''' Generate cache of available packages '''
# We generate this cache only if the repos were just freshed or if the
# cache file doesn't exist
fresh = False
for repo in self.base.repos.iter_enabled():
if repo.metadata is not None and repo.metadata.fresh:
# One fresh repo is enough to cause a regen of the cache
fresh = True
break
if not os.path.exists(self.cache_file) or fresh:
try:
with sqlite3.connect(self.cache_file) as conn:
self._out('Generating completion cache...')
cur = conn.cursor()
cur.execute(
"create table if not exists available (pkg TEXT)")
cur.execute(
"create unique index if not exists "
"pkg_available ON available(pkg)")
cur.execute("delete from available")
avail_pkgs = self.base.sack.query().available()
avail_pkgs_insert = [[str(x)] for x in avail_pkgs if x.arch != "src"]
cur.executemany("insert or ignore into available values (?)",
avail_pkgs_insert)
conn.commit()
except sqlite3.OperationalError as e:
self._out("Can't write completion cache: %s" % ucd(e))
def transaction(self):
''' Generate cache of installed packages '''
if not self.transaction:
return
try:
with sqlite3.connect(self.cache_file) as conn:
self._out('Generating completion cache...')
cur = conn.cursor()
cur.execute("create table if not exists installed (pkg TEXT)")
cur.execute(
"create unique index if not exists "
"pkg_installed ON installed(pkg)")
cur.execute("delete from installed")
inst_pkgs = dnf.sack._rpmdb_sack(self.base).query().installed()
inst_pkgs_insert = [[str(x)] for x in inst_pkgs if x.arch != "src"]
cur.executemany("insert or ignore into installed values (?)",
inst_pkgs_insert)
conn.commit()
except sqlite3.OperationalError as e:
self._out("Can't write completion cache: %s" % ucd(e))
Zerion Mini Shell 1.0