forked from sig_core/toolkit
57 lines
1.4 KiB
Python
Executable File
57 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# label <label@rockylinux.org>
|
|
import datetime
|
|
import sys
|
|
import dnf
|
|
import dnf.exceptions
|
|
|
|
# pylint: disable=unnecessary-lambda-assignment
|
|
now = datetime.datetime.today().strftime("%m-%d-%Y %T")
|
|
|
|
class DnfQuiet(dnf.Base):
|
|
"""
|
|
DNF object
|
|
|
|
This is in the event we need special functions
|
|
"""
|
|
def __init__(self):
|
|
dnf.Base.__init__(self)
|
|
|
|
def main():
|
|
"""
|
|
Main run
|
|
"""
|
|
dnfobj = DnfQuiet()
|
|
releasever = dnfobj.conf.releasever
|
|
try:
|
|
dnfobj.read_all_repos()
|
|
# pylint: disable=bare-except
|
|
except:
|
|
print(f'[-] {now} -> Could not read repos', file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
rocky_default_repos = {
|
|
'8': ['baseos', 'appstream', 'extras'],
|
|
'9': ['baseos', 'appstream', 'extras']
|
|
}.get(releasever, None)
|
|
|
|
if not rocky_default_repos:
|
|
print(f'[-] {now} -> Not a Rocky Linux system')
|
|
sys.exit(1)
|
|
|
|
print(f'[-] {now} -> Checking if non-default repo is enabled')
|
|
_not_allowed=False
|
|
for repo in list(dnfobj.repos.iter_enabled()):
|
|
if not repo.id in rocky_default_repos:
|
|
print(f'[-] {now} -> {repo.id} is enabled and should be disabled')
|
|
_not_allowed=True
|
|
if _not_allowed:
|
|
print(f'[-] {now} -> FAIL - There are extra repos enabled')
|
|
sys.exit(1)
|
|
|
|
print(f'[-] {now} -> PASS')
|
|
sys.exit(0)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|