InPanel 1.2.1__py3-none-any.whl

inpanel/mod/disk.py

@@ -0,0 +1,322 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2017-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of the (new) BSD License.
+# The full license can be found in 'LICENSE'.
+
+'''Module for disk Management'''
+
+import os
+from pathlib import Path
+from shlex import split
+
+import pexpect
+
+from .config import readconfig, writeconfig
+from . import server
+
+
+def add(disk, size=''):
+    """Add a new partition on a disk.
+
+    If the size exceed the max available space the disk left, then the
+    new partition will be created with the left space.
+
+    A disk can have 4 partitions at max.
+
+    True will return if create successfully, or else False will return.
+
+    Example:
+    fdisk.add('/dev/sdb')   # use all of the space
+    fdisk.add('/dev/sdb', '5G') # create a partition with at most 5G space
+    """
+    try:
+        cmd = split('fdisk \'%s\'' % disk)
+    except:
+        return False
+
+    child = pexpect.spawn(cmd[0], cmd[1:])
+    i = child.expect(['(m for help)', 'Unable to open'])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    rt = True
+    partno_found = False
+    partno = 1
+    while not partno_found:
+        child.sendline('n')
+        i = child.expect(
+            ['primary partition', 'You must delete some partition'])
+        if i == 1:
+            break
+        child.sendline('p')
+
+        i = child.expect(['Partition number', 'Selected partition'])
+        if i == 0:
+            child.sendline('%d' % partno)
+
+        i = child.expect(['First cylinder', '(m for help)'])
+        if i == 0:
+            partno_found = True
+        partno += 1
+        if partno > 4:
+            break
+
+    if not partno_found:
+        rt = False
+
+    if rt:
+        child.sendline('')
+        child.expect('Last cylinder')
+        child.sendline('+%s' % size)
+        i = child.expect(
+            ['(m for help)', 'Value out of range', 'Last cylinder'])
+        if i == 1:
+            child.sendline('')
+            child.expect('(m for help)')
+        elif i == 2:    # wrong size input
+            child.sendline('')
+            child.expect('(m for help)')
+            rt = False
+
+    if rt:
+        child.sendline('w')
+    else:
+        child.sendline('q')
+
+    if child.isalive():
+        child.wait()
+    return rt
+
+
+def delete(partition):
+    """Delete a partition.
+
+    True will return if delete successfully, or else False will return.
+
+    Example:
+    fdisk.delete('/dev/sdb1')
+    """
+    disk = partition[:-1]
+    partno = partition[-1:]
+    try:
+        cmd = split('fdisk \'%s\'' % disk)
+    except:
+        return False
+
+    child = pexpect.spawn(cmd[0], cmd[1:])
+    i = child.expect(['(m for help)', 'Unable to open'])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline('d')
+
+    rt = True
+    i = child.expect([
+        'Partition number',
+        'Selected partition %s' % partno,
+        'No partition is defined yet',
+        pexpect.TIMEOUT
+    ], timeout=1)
+    if i == 0:
+        child.sendline(partno)
+    elif i == 2 or i == 3:
+        rt = False
+
+    if rt:
+        i = child.expect(['(m for help)', 'has empty type'])
+        if i == 0:
+            child.sendline('w')
+        elif i == 1:
+            rt = False
+
+    if not rt:
+        child.sendline('q')
+
+    if child.isalive():
+        child.wait()
+    return rt
+
+
+def scan(disk, size=''):
+    """Rescan partitions on a disk.
+
+    True will return if scan successfully, or else False will return.
+
+    Example:
+    fdisk.scan('/dev/sdb')
+    """
+    try:
+        cmd = split('fdisk \'%s\'' % disk)
+    except:
+        return False
+
+    child = pexpect.spawn(cmd[0], cmd[1:])
+    i = child.expect(['(m for help)', 'Unable to open'])
+    if i == 1:
+        child.wait()
+        return False
+
+    child.sendline('w')
+    i = child.expect([
+        'The kernel still uses the old table',
+        pexpect.TIMEOUT,
+        pexpect.EOF
+    ], timeout=1)
+    if i == 0:
+        rt = False
+    else:
+        rt = True
+
+    if child.isalive():
+        child.wait()
+    return rt
+
+
+def handle_fdisk(action, devname, size='', unit=''):
+    """Handle fdisk operations.
+
+    Args:
+        action: 'add', 'delete', or 'scan'
+        devname: device name (e.g., 'sdb')
+        size: partition size (e.g., '5')
+        unit: 'M' or 'G'
+
+    Returns:
+        dict with 'code' and 'msg' keys
+    """
+    if action == 'add':
+        if unit not in ('M', 'G'):
+            return {'code': -1, 'msg': '错误的分区大小！'}
+
+        if size == '':
+            size = None  # use whole left space
+        else:
+            try:
+                size = float(size)
+            except:
+                return {'code': -1, 'msg': '错误的分区大小！'}
+
+            if unit == 'G' and size - int(size) > 0:
+                size *= 1024
+                unit = 'M'
+            size = '%d%s' % (round(size), unit)
+
+        if add('/dev/%s' % devname, size):
+            return {'code': 0, 'msg': '在 %s 设备上创建分区成功！' % devname}
+        else:
+            return {'code': -1, 'msg': '在 %s 设备上创建分区失败！' % devname}
+
+    elif action == 'delete':
+        if delete('/dev/%s' % devname):
+            fstab(devname, {'devname': devname, 'mount': None})
+            return {'code': 0, 'msg': '分区 %s 删除成功！' % devname}
+        else:
+            return {'code': -1, 'msg': '分区 %s 删除失败！' % devname}
+
+    elif action == 'scan':
+        if scan('/dev/%s' % devname):
+            return {'code': 0, 'msg': '扫描设备 %s 的分区成功！' % devname}
+        else:
+            return {'code': -1, 'msg': '扫描设备 %s 的分区失败！' % devname}
+
+    else:
+        return {'code': -1, 'msg': '未定义的操作！'}
+
+
+def _read_fstab(line, **params):
+    """Read fstab config for a specific device.
+    
+    Args:
+        line: a line from /etc/fstab
+        params: contains 'devname' parameter
+    
+    Returns:
+        dict with config or None
+    """
+    if not line or line.startswith('#'):
+        return
+    fields = line.split()
+    dev = fields[0]
+    config = {
+        'dev':    fields[0],
+        'mount':  fields[1],
+        'fstype': fields[2],
+    }
+    if dev.startswith('/dev/'):
+        try:
+            devlink = os.readlink(dev)
+            dev = str(Path(devlink).parent.joinpath(dev).resolve())
+        except:
+            pass
+        dev = dev.replace('/dev/', '')
+        if dev == params['devname']:
+            return config
+    elif dev.startswith('UUID='):
+        uuid = dev.replace('UUID=', '')
+        partinfo = server.ServerInfo.partinfo(devname=params['devname'])
+        if partinfo['uuid'] == uuid:
+            return config
+
+
+def _write_fstab(line, **params):
+    """Write fstab config for a specific device.
+    
+    Args:
+        line: a line from /etc/fstab (None for new line)
+        params: contains 'devname' and 'config' parameters
+    
+    Returns:
+        formatted line string or None to remove
+    """
+    config = params['config']
+    if not 'mount' in config or config['mount'] is None:
+        return None  # remove line
+    if line is None:  # new line
+        return '/dev/%s %s                %s    defaults        1 2' % \
+            (params['devname'], config['mount'], config['fstype'])
+    else:  # update existing line
+        fields = line.split()
+        return '%s %s                %s    %s        %s %s' % \
+            (fields[0], config['mount'],
+             'fstype' in config and config['fstype'] or fields[2],
+             fields[3], fields[4], fields[5])
+
+
+def fstab(devname, config=None):
+    """Read or write config from /etc/fstab.
+
+    Args:
+        devname: device name (e.g., 'sda1')
+        config: None for read, dict for write/remove
+    
+    Returns:
+        config dict on read, boolean on write
+    """
+    cfgfile = '/etc/fstab'
+    if config is None:
+        # read config
+        return readconfig(cfgfile, _read_fstab, devname=devname)
+    else:
+        # write or remove config
+        return writeconfig(cfgfile, _read_fstab, _write_fstab,
+                           devname=devname, config=config)
+
+
+if __name__ == '__main__':
+    # !!!!!!!!!!! DANGEROUS TESTING !!!!!!!!!!!
+    # print('* Add partition to sdb with 5G:',)
+    # print(add('/dev/sdb', '5G'))
+
+    # print('* Delete partition /dev/sdb1:')
+    # print(delete('/dev/sdb1'))
+
+    print('* Rescan partitions of /dev/sdb:')
+    print(scan('/dev/sdb'))

inpanel/mod/dnf.py

@@ -0,0 +1,239 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2020-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of The New BSD License.
+# The full license can be found in 'LICENSE'.
+'''Module for DNF Repository Management'''
+
+import os
+from pathlib import Path
+
+from .. import base
+from . import config
+from . import file
+from ..dnf import dnf_reporpms
+
+config_path = '/etc/yum.repos.d'
+
+
+def get_list():
+    '''get repo list'''
+    res = []
+    if base.kernel_name in ('Linux'):
+        d = str(Path(config_path))
+        if not Path(d).exists() or not Path(d).is_dir():
+            return None
+        items = sorted(os.listdir(d))
+        return items if len(items) > 0 else []
+    elif base.kernel_name in ('Darwin'):
+        return None
+    else:
+        return None
+
+
+def item_exists(repo):
+    '''exists'''
+    return Path(str(Path(config_path) / repo)).exists()
+
+
+def get_item(repo):
+    '''get repo config'''
+    if not repo:
+        return None
+    repo_file = str(Path(config_path) / repo)
+    if Path(repo_file).exists():
+        _config = config.Config(repo_file)
+        return _config.get_config()
+    return None
+
+
+def set_item(repo, data):
+    '''set repo config'''
+    if not repo:
+        return None
+    if not data:
+        return None
+    repo_file = str(Path(config_path) / repo)
+    if Path(repo_file).exists():
+        _config = config.Config(repo_file, data)
+        return True
+    else:
+        return False
+
+
+def add_item(repo, data):
+    '''add repo config'''
+    if not repo:
+        return None
+    if not data:
+        return None
+    repo_file = str(Path(config_path) / repo)
+    if Path(repo_file).exists():
+        return False
+    else:
+        _config = config.Config(repo_file, data)
+        return True
+
+
+def del_item(repo):
+    '''delete repo file'''
+    if not repo:
+        return None
+    repo_file = str(Path(config_path) / repo)
+    return file.delete(repo_file)
+
+
+def get_repo_release(os_versint, os_name, arch):
+    '''install release'''
+    cmds = []
+    if os_versint >= 8:
+        if os_name in ('centos', 'almalinux', 'rocky', 'rhel', 'fedora'):
+            cmds.append('dnf install -y %s-release' % os_name)
+    else:
+        for rpm in dnf_reporpms['base'][os_versint][arch]:
+            cmds.append('rpm -U %s' % rpm)
+    return cmds
+
+
+def get_repo_epel(os_versint, os_name, arch):
+    '''install epel'''
+    cmds = []
+    if os_versint >= 8:
+        cmds.append('dnf install -y epel-release')
+    else:
+        for rpm in dnf_reporpms['epel'][os_versint][arch]:
+            cmds.append('rpm -U %s' % rpm)
+    return cmds
+
+
+def get_repo_powertools(os_versint, os_name, arch):
+    '''install powertools/crb'''
+    cmds = []
+    if os_versint == 8:
+        cmds.append('dnf install -y dnf-plugins-core')
+        cmds.append('dnf config-manager --set-enabled powertools')
+    elif os_versint >= 9:
+        cmds.append('dnf install -y dnf-plugins-core')
+        cmds.append('dnf config-manager --set-enabled crb')
+    return cmds
+
+
+def get_repo_remi(os_versint, os_name, arch):
+    '''install remi repository'''
+    cmds = []
+    if os_versint >= 8:
+        cmds.append('dnf install -y https://rpms.remirepo.net/enterprise/remi-release-%d.rpm' % os_versint)
+    return cmds
+
+
+def web_handler(context):
+    '''Handle web requests for DNF repository management'''
+    action = context.get('action', '')
+    repo = context.get('repo', '')
+    
+    if action == 'list':
+        items = get_list()
+        if items is None:
+            return {'code': -1, 'msg': '获取配置失败！'}
+        else:
+            return {'code': 0, 'msg': '', 'data': items}
+    
+    elif action == 'item':
+        if not repo:
+            return {'code': -1, 'msg': '配置文件不能为空！'}
+        data = get_item(repo)
+        if data is None:
+            return {'code': -1, 'msg': '配置文件不存在！'}
+        else:
+            return {'code': 0, 'msg': '', 'data': data}
+    
+    elif action == 'add':
+        if not repo:
+            return {'code': -1, 'msg': '配置文件不能为空！'}
+        serverid = context.get('serverid', '')
+        if not serverid:
+            return {'code': -1, 'msg': '仓库标识ID不能为空！'}
+        name = context.get('name', '')
+        if not name:
+            return {'code': -1, 'msg': '仓库名称不能为空！'}
+        baseurl = context.get('baseurl', '')
+        if not baseurl:
+            return {'code': -1, 'msg': '仓库路径不能为空！'}
+        
+        enabled = context.get('enabled', True)
+        gpgcheck = context.get('gpgcheck', False)
+        
+        data = {
+            serverid: {
+                'name': name,
+                'enabled': 0 if not enabled else 1,
+                'baseurl': baseurl,
+                'gpgcheck': 0 if not gpgcheck else 1,
+                'gpgkey': ''
+            }
+        }
+        
+        if item_exists(repo):
+            return {'code': -1, 'msg': '配置文件已存在！'}
+        if add_item(repo, data) is True:
+            return {'code': 0, 'msg': '配置添加成功！'}
+        else:
+            return {'code': -1, 'msg': '配置添加失败！'}
+    
+    elif action == 'edit':
+        if not repo:
+            return {'code': -1, 'msg': '配置文件不能为空！'}
+        serverid = context.get('serverid', '')
+        if not serverid:
+            return {'code': -1, 'msg': '仓库标识ID不能为空！'}
+        name = context.get('name', '')
+        if not name:
+            return {'code': -1, 'msg': '仓库名称不能为空！'}
+        baseurl = context.get('baseurl', '')
+        if not baseurl:
+            return {'code': -1, 'msg': '仓库路径不能为空！'}
+        
+        enabled = context.get('enabled', True)
+        gpgcheck = context.get('gpgcheck', False)
+        
+        data = {
+            serverid: {
+                'name': name,
+                'enabled': 0 if not enabled else 1,
+                'baseurl': baseurl,
+                'gpgcheck': 0 if not gpgcheck else 1,
+                'gpgkey': ''
+            }
+        }
+        
+        if not item_exists(repo):
+            return {'code': -1, 'msg': '配置文件不存在！'}
+        if set_item(repo, data) is True:
+            return {'code': 0, 'msg': '配置修改成功！'}
+        else:
+            return {'code': -1, 'msg': '配置修改失败！'}
+    
+    elif action == 'del':
+        if not repo:
+            return {'code': -1, 'msg': '配置文件不能为空！'}
+        if not item_exists(repo):
+            return {'code': -1, 'msg': '配置文件不存在！'}
+        if del_item(repo) is True:
+            return {'code': 0, 'msg': '配置文件已移入回收站！'}
+        else:
+            return {'code': -1, 'msg': '删除失败！'}
+    
+    else:
+        return {'code': -1, 'msg': '未定义的操作！'}
+
+
+if __name__ == '__main__':
+    l = get_list()
+    print(l)
+    if l:
+        i = l[0]
+        print(i)
+        c = get_item(i)
+        print(c)