Windowsのtreeをpathに変換するコードを書いてみた。

PCを初期化したとき、初期の構造を記憶するためにディレクトリ構造を取得したくて取得したのだが、
dir C:\\* /s /b /adで取得すればいいものを
treeで取得してしまった。

でも今更もっかい初期化とかできないし。どうしよう。
そうだ、唯一使えるPythonを使ってPath化しよう！

ということで書いた。

コード

ちなみに私はインデントにタブを使っている。タブが好きだからだ。クソコードごめんね

import os


class DirectoryTreeChangeToPath:

	def __init__(self, path):
		paths = self.shaping(self.get_file_content(path))
		with open(".\\changed_directory_tree_to_path.txt","x") as f:
			for i in paths:
				f.write(i + "\n")

	@staticmethod
	def get_file_content(path, mode="r", encoding='utf-8'):
		if os.path.exists(path):
			with open(path, mode=mode, encoding=encoding) as f:
				for i in f:
					yield i
		else:
			raise FileNotFoundError

	@staticmethod
	def shaping(default_dir_tree_data):
		directory = [next(default_dir_tree_data)[:-2]]
		for i in default_dir_tree_data:
			i = i.replace("\n", "", -1)
			i = DirectoryTreeChangeToPath.pipe_count(i)
			DirectoryTreeChangeToPath.merge(directory,i)
			yield "\\".join(directory)

	@staticmethod
	def merge(subject_,object_):
		if len(subject_) > object_[0]:
			del subject_[object_[0]:]
		return subject_.append(object_[1])

	@staticmethod		
	def pipe_count(string):
		indent = 0
		while True:
			if string[:2] in ("└─","├─"):
				string = string[2:]
				indent += 1
				break
			elif string[:3] == "│  ":
				string = string[3:]
				indent += 1
			elif string[:4] == "    ":
				string = string[4:]
				indent += 1
		return indent,string


def main():
	os.chdir(os.path.dirname(__file__))
	DirectoryTreeChangeToPath(".\\infomation\\default_dir_tree_data")


main()

書いてる間の思考

パイプをただのインデントとしてみたら簡単に数値に置換できた。