이 Python/Xonsh 스크립트(bash에서 변환됨)를 어떻게 개선할 수 있습니까?

Question

이것은 내가 직접 생각해낸 것 중 단연 최고입니다.

저는 두 개의 함수를 작성하여 modtest라는 디렉터리에 저장했습니다(그리고 파일을 ffutils.py로 저장했습니다).
touch modtest/__init__.py
바라보다https://stackoverflow.com/a/33770042이에 대해 자세히 알아보세요.

import sys
from pathlib import Path
import re

def find_profile_dirs(user=None, debug=False):
  profiles = []
  if user is None:
    base_path = Path('~/.mozilla/firefox').expanduser()
  else:
    base_path = Path('/home') / user / '.mozilla/firefox'
  if not base_path.is_dir():
    print("ERROR:", base_path, "does not exist.")
  elif debug:
    print("OK: processing", base_path)
  ini_path = Path(base_path / 'profiles.ini')
  if not ini_path.is_file():
    print("ERROR:", ini_path, "cannot be opened.")
  elif debug:
    print("OK: opening", ini_path)
  pro_path_re = re.compile(r'^Path=(\w+)$')
  with open(ini_path, "r") as profiles_ini:
    for any_line in profiles_ini:
      m_path = pro_path_re.match(any_line)
      if not m_path is None:
        p_path = m_path.group(1)
        if debug:
          print("p_path:", p_path)
        profile_path = Path(base_path / p_path)
        if debug:
          print("profile_path:", profile_path)
        if profile_path.is_dir():
          profiles.append(profile_path)
          print("The profile path is", profile_path)
  return profiles


# profiles_paths is a list of Paths of all Firefox profiles to process
def process_userjs(searchExp, replaceExp, profiles_paths, debug=False):
  match_result = None
  search_regex = re.compile(searchExp, re.M)

  trgt_file='user.js'
  for profile_path in profiles_paths:
    user_js_path = Path(profile_path / trgt_file)
    if Path(user_js_path).is_file():
      if debug:
        print("OK: valid file:", user_js_path)
      with open(user_js_path, 'r+') as file_userjs:
        file_contents = file_userjs.read()
        match_result = search_regex.search(file_contents)
        if match_result is None:
          file_contents = file_contents + '\n' + replaceExp
          if debug:
            print('No match: added new line to end of file contents:', replaceExp)
        else:
          file_contents = search_regex.sub(replaceExp, file_contents)
          if debug:
            print('Match found. Modified existing line in file.')

        file_userjs.seek(0)
        file_userjs.truncate()
        file_userjs.write(file_contents)
    elif debug:
      print("SKIPPED: invalid file:", user_js_path)

그런 다음 다음과 같은 기능을 사용했습니다.

import sys
sys.path = ['/path/to/parent/of/modtest'] + sys.path # if needed
from modtest import ffutils

my_profiles = ffutils.find_profile_dirs()

srchexp = r'^/*?user_pref\("browser\.ctrlTab\.sortByRecentlyUsed.*$'
replexp = 'user_pref("browser.ctrlTab.sortByRecentlyUsed", true);'
ffutils.process_userjs(srchexp, replexp, my_profiles, True)

물론 다른 검색을 사용하여 이를 반복하고 문자열을 바꿀 수도 있습니다.

나는 이것을 개선할 수 있는 방법이 있다고 확신합니다. 나는 다른 답변을 받아들이겠지만, 다른 답변이 없을 경우를 대비해 여기에 내 답변을 넣습니다.

내 코드에는 몇 가지 가정이 있습니다. 한 가지 큰 문제는 모든 Firefox 프로필 경로가 상대적이라고 가정한다는 것입니다. 이것이 유효한 가정이 아닌 경우에는profiles.ini에 대한 추가 처리가 필요합니다.

Answer 1