75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
# This is a sample Python script.
|
|
|
|
# Press ⌃R to execute it or replace it with your code.
|
|
# Press Double ⇧ to search everywhere for classes, files, tool windows, actions, and settings.
|
|
|
|
import argparse
|
|
import os
|
|
import tarfile
|
|
|
|
|
|
class TarRoot:
|
|
def __init__(self, arguments):
|
|
self.debug = arguments.debug
|
|
self.target_dir = arguments.target_dir
|
|
self.tar = tarfile.open(arguments.target_tar, "w:gz")
|
|
|
|
def debug_print(self, *arguments):
|
|
if self.debug:
|
|
output = ""
|
|
for arg in arguments:
|
|
output += repr(arg)
|
|
print("DEBUG: " + output)
|
|
|
|
def run(self):
|
|
start_dir = os.getcwd()
|
|
self.debug_print("entering target dir: ", self.target_dir)
|
|
os.chdir(self.target_dir)
|
|
self._run('.')
|
|
self.debug_print("returning to original dir: ", start_dir)
|
|
os.chdir(start_dir)
|
|
|
|
def _append_dir(self, entry, *parent):
|
|
path = entry.name
|
|
if len(parent) > 0:
|
|
path = parent[0] + os.sep + path
|
|
ti = tarfile.TarInfo(path)
|
|
ti.type = tarfile.DIRTYPE
|
|
stat_result = entry.stat(follow_symlinks=False)
|
|
ti.mode = stat_result.st_mode
|
|
ti.uid = stat_result.st_uid
|
|
ti.gid = stat_result.st_gid
|
|
ti.mtime = stat_result.st_mtime
|
|
self.debug_print("writing tar info ", ti)
|
|
self.tar.addfile(ti)
|
|
|
|
def _append_file(self, entry):
|
|
pass
|
|
|
|
def _run(self, running_dir):
|
|
symlinks = []
|
|
with os.scandir(running_dir) as it:
|
|
for entry in it:
|
|
if entry.is_symlink():
|
|
self.debug_print("appending symlink ", entry)
|
|
symlinks.append(entry)
|
|
elif entry.is_dir():
|
|
self.debug_print("appending directory ", entry)
|
|
self._append_dir(entry, running_dir) if running_dir != "." else self._append_dir(entry)
|
|
self.debug_print("going recursive")
|
|
self._run(running_dir + os.sep + entry.name) if running_dir != "." else self._run(entry.name)
|
|
else:
|
|
self.debug_print("appending file ", entry)
|
|
self._append_file(entry)
|
|
|
|
|
|
# Press the green button in the gutter to run the script.
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--debug", help="debug flag", action="store_true")
|
|
parser.add_argument("target_dir")
|
|
parser.add_argument("target_tar")
|
|
args = parser.parse_args()
|
|
tr = TarRoot(args)
|
|
tr.run()
|