dirutility.open.clazz.DirBase.__init__() - Code Metrics - sfneal/dirutility - Measure and Improve Code Quality continuously with Scrutinizer

dirutility.open.clazz.DirBase.init() A
last analyzed 2022-05-24 16:44 UTC

↳ Parent: dirutility.open.clazz

Complexity

Conditions

Size

Total Lines	3
Code Lines	3

Duplication

Lines	3
Ratio	100 %

Importance

Changes

Metric	Value
cc	1
eloc	3
nop	3
dl	3
loc	3
rs	10
c	0
b	0
f	0

from __future__ import annotations

import abc
import os
import tarfile
from typing import Union, Dict, TextIO, List
from zipfile import PyZipFile

from dirutility.error import InvalidFileNameError, InvalidDirStructureError


class DirStructure:


    def __init__(self, **paths: Dict[str, Union[dict, str, TextIO, None]]):
        self._content = {}
        for pathname, _c in paths.items():
            if isinstance(_c, dict):
                self._content[pathname] = DirStructure(**_c)
            else:
                self._content[pathname] = _c

    @property
    def content(self):
        return self._content.copy()

    @property
    def dirsname(self) -> List[str]:
        res = list(self._content.keys())
        res.sort()
        return res

    def _is_filename_valid(self, filename):
        if isinstance(filename, str) and filename.strip():
            return True
        else:
            return False

    def validate_filename(self, filename):
        if not self._is_filename_valid(filename):
            raise InvalidFileNameError(filename)

    def create(self, prefix_dir: str):
        for filename, _c in self._content.items():
            self.validate_filename(filename)
            path = '{}/{}'.format(prefix_dir, filename)
            if _c is None:
                fd = open(path, 'w')
                fd.close()
            if isinstance(_c, str):
                with open(path, 'w') as fd:
                    fd.write(_c)
            elif isinstance(_c, dict):
                os.mkdir(path)
            elif isinstance(_c, DirStructure):
                os.mkdir(path)
                _c.create(path)
            else:
                with open(path, 'w') as fd:
                    for _l in _c:
                        fd.write(_l)


class DirBase(metaclass=abc.ABCMeta):


    def __init__(self, dirpath: str, structure: Union[DirStructure, None] = None):
        self._dirpath = dirpath
        self._structure = structure

    @property
    def dirname(self) -> str:
        return self.dirpath.rsplit('/', 1)[1]

    @property
    def dirpath(self) -> str:
        return self._dirpath

    @property
    def dirparent(self) -> str:
        return self.dirpath.rsplit('/', 1)[0]

    def dirs(self, relative=True) -> List[str]:
        if relative:
            return self._structure.dirsname
        else:
            return ['{}/{}'.format(self._dirpath, _name) for _name in self._structure.dirsname]

    def create_structure(self, **paths: Dict[str, Union[dict, str, TextIO, None]]):
        self._structure = DirStructure(**paths)
        self._structure.create(self._dirpath)


class TempDir(DirBase):
    pass


class TarFile(object):


    def __init__(self, filepath: str, tempdir: TempDir, mode: str = 'w:gz'):

        self._mode = mode.strip()
        _ar = self._mode.split(':')
        if len(_ar) > 1 and _ar[-1].strip():
            self._compress_type = _ar[-1].strip()
        else:
            self._compress_type = None
        self._open_mode = _ar[0]
        self._filepath = filepath.strip()
        self._filename = self._filepath.rsplit('/', 1)[-1]
        self._filename_without_suffix = self._filename.rsplit('.', 1)[0]
        self._validate_filename(self._filepath)
        self._tempdir = tempdir

    def _is_filename_valid(self, filename):
        if filename.endswith(self._compress_type):
            return True
        else:
            return False

    def _validate_filename(self, filename):
        if not self._is_filename_valid(filename):
            raise InvalidFileNameError(filename)

    @property
    def filepath(self) -> str:
        return self._filepath

    def filename(self, with_suffix=True) -> str:
        if with_suffix:
            return self._filename
        else:
            return self._filename_without_suffix

    @property
    def tempdir(self) -> TempDir:
        return self._tempdir

    def create_archive(self, withdir: bool = False):
        with tarfile.open(self._filepath, mode=self._mode) as tar:
            for dirpath, dirname in zip(self._tempdir.dirs(relative=False), self._tempdir.dirs()):
                tar.add(dirpath, arcname=dirname, recursive=True)


class ZipFile(TarFile):


    def __init__(self, filepath: str, tempdir: TempDir, mode: str = 'w:zip'):
        super(ZipFile, self).__init__(filepath, tempdir, mode)

    def create_archive(self, withdir: bool = False):
        zip_file = PyZipFile(self._filepath, mode=self._open_mode)
        for root, dirs, files in os.walk(self._tempdir.dirpath):
            if files:
                for file in files:
                    _fp = os.path.join(root, file)
                    if withdir:
                        zip_file.write(_fp,
                                       arcname='{}/{}'.format(self.filename(with_suffix=False),
                                                              _fp.replace(self._tempdir.dirpath, '')))
                    else:
                        zip_file.write(_fp, arcname=_fp.replace(self._tempdir.dirpath, ''))

            else:
                if withdir:
                    zip_file.write(root,
                                   arcname='{}/{}'.format(self.filename(with_suffix=False),
                                                          root.replace(self._tempdir.dirpath, '')))
                else:
                    zip_file.write(root, arcname=root.replace(self._tempdir.dirpath, ''))


1		from __future__ import annotations
2
3		import abc
4		import os
5		import tarfile
6		from typing import Union, Dict, TextIO, List
7		from zipfile import PyZipFile
8
9		from dirutility.error import InvalidFileNameError, InvalidDirStructureError
10
11
12	View Code Duplication	class DirStructure:
		0 ignored issues – show Duplication introduced 2022-05-24 16:39 UTC by Report Bug Copy Issue Report This code seems to be duplicated in your project. Loading history...
13
14		def __init__(self, **paths: Dict[str, Union[dict, str, TextIO, None]]):
15		self._content = {}
16		for pathname, _c in paths.items():
17		if isinstance(_c, dict):
18		self._content[pathname] = DirStructure(**_c)
19		else:
20		self._content[pathname] = _c
21
22		@property
23		def content(self):
24		return self._content.copy()
25
26		@property
27		def dirsname(self) -> List[str]:
28		res = list(self._content.keys())
29		res.sort()
30		return res
31
32		def _is_filename_valid(self, filename):
33		if isinstance(filename, str) and filename.strip():
34		return True
35		else:
36		return False
37
38		def validate_filename(self, filename):
39		if not self._is_filename_valid(filename):
40		raise InvalidFileNameError(filename)
41
42		def create(self, prefix_dir: str):
43		for filename, _c in self._content.items():
44		self.validate_filename(filename)
45		path = '{}/{}'.format(prefix_dir, filename)
46		if _c is None:
47		fd = open(path, 'w')
48		fd.close()
49		if isinstance(_c, str):
50		with open(path, 'w') as fd:
51		fd.write(_c)
52		elif isinstance(_c, dict):
53		os.mkdir(path)
54		elif isinstance(_c, DirStructure):
55		os.mkdir(path)
56		_c.create(path)
57		else:
58		with open(path, 'w') as fd:
59		for _l in _c:
60		fd.write(_l)
61
62
63	View Code Duplication	class DirBase(metaclass=abc.ABCMeta):
		0 ignored issues – show Duplication introduced 2022-05-24 16:39 UTC by Report Bug Copy Issue Report This code seems to be duplicated in your project. Loading history...
64
65		def __init__(self, dirpath: str, structure: Union[DirStructure, None] = None):
66		self._dirpath = dirpath
67		self._structure = structure
68
69		@property
70		def dirname(self) -> str:
71		return self.dirpath.rsplit('/', 1)[1]
72
73		@property
74		def dirpath(self) -> str:
75		return self._dirpath
76
77		@property
78		def dirparent(self) -> str:
79		return self.dirpath.rsplit('/', 1)[0]
80
81		def dirs(self, relative=True) -> List[str]:
82		if relative:
83		return self._structure.dirsname
84		else:
85		return ['{}/{}'.format(self._dirpath, _name) for _name in self._structure.dirsname]
86
87		def create_structure(self, **paths: Dict[str, Union[dict, str, TextIO, None]]):
88		self._structure = DirStructure(**paths)
89		self._structure.create(self._dirpath)
90
91
92		class TempDir(DirBase):
93		pass
94
95
96	View Code Duplication	class TarFile(object):
		0 ignored issues – show Duplication introduced 2022-05-24 16:39 UTC by Report Bug Copy Issue Report This code seems to be duplicated in your project. Loading history...
97
98		def __init__(self, filepath: str, tempdir: TempDir, mode: str = 'w:gz'):
99
100		self._mode = mode.strip()
101		_ar = self._mode.split(':')
102		if len(_ar) > 1 and _ar[-1].strip():
103		self._compress_type = _ar[-1].strip()
104		else:
105		self._compress_type = None
106		self._open_mode = _ar[0]
107		self._filepath = filepath.strip()
108		self._filename = self._filepath.rsplit('/', 1)[-1]
109		self._filename_without_suffix = self._filename.rsplit('.', 1)[0]
110		self._validate_filename(self._filepath)
111		self._tempdir = tempdir
112
113		def _is_filename_valid(self, filename):
114		if filename.endswith(self._compress_type):
115		return True
116		else:
117		return False
118
119		def _validate_filename(self, filename):
120		if not self._is_filename_valid(filename):
121		raise InvalidFileNameError(filename)
122
123		@property
124		def filepath(self) -> str:
125		return self._filepath
126
127		def filename(self, with_suffix=True) -> str:
128		if with_suffix:
129		return self._filename
130		else:
131		return self._filename_without_suffix
132
133		@property
134		def tempdir(self) -> TempDir:
135		return self._tempdir
136
137		def create_archive(self, withdir: bool = False):
138		with tarfile.open(self._filepath, mode=self._mode) as tar:
139		for dirpath, dirname in zip(self._tempdir.dirs(relative=False), self._tempdir.dirs()):
140		tar.add(dirpath, arcname=dirname, recursive=True)
141
142
143	View Code Duplication	class ZipFile(TarFile):
		0 ignored issues – show Duplication introduced 2022-05-24 16:39 UTC by Report Bug Copy Issue Report This code seems to be duplicated in your project. Loading history...
144
145		def __init__(self, filepath: str, tempdir: TempDir, mode: str = 'w:zip'):
146		super(ZipFile, self).__init__(filepath, tempdir, mode)
147
148		def create_archive(self, withdir: bool = False):
149		zip_file = PyZipFile(self._filepath, mode=self._open_mode)
150		for root, dirs, files in os.walk(self._tempdir.dirpath):
151		if files:
152		for file in files:
153		_fp = os.path.join(root, file)
154		if withdir:
155		zip_file.write(_fp,
156		arcname='{}/{}'.format(self.filename(with_suffix=False),
157		_fp.replace(self._tempdir.dirpath, '')))
158		else:
159		zip_file.write(_fp, arcname=_fp.replace(self._tempdir.dirpath, ''))
160
161		else:
162		if withdir:
163		zip_file.write(root,
164		arcname='{}/{}'.format(self.filename(with_suffix=False),
165		root.replace(self._tempdir.dirpath, '')))
166		else:
167		zip_file.write(root, arcname=root.replace(self._tempdir.dirpath, ''))
168

sfneal / dirutility

dirutility.open.clazz.DirBase.__init__() A last analyzed 2022-05-24 16:44 UTC

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like

dirutility.open.clazz.DirBase.init() A
last analyzed 2022-05-24 16:44 UTC