diff --git a/core/utils.py b/core/utils.py index 269d719..7259bae 100644 --- a/core/utils.py +++ b/core/utils.py @@ -5,7 +5,7 @@ import re import resource import random import logging -from subprocess import call +import subprocess from urllib.parse import urljoin from decimal import Decimal, getcontext @@ -79,13 +79,61 @@ def GetListDir(sdir): res.append((f, ff, os.path.isdir(ff))) return res +def write_and_commit(files, message): + """Writes the content to the filepath and adds and commits the file to git. If this fails, a WriteAndCommitError is raised.""" + git = settings.GIT + try: + for filepath, content, encoding in files: + cwd = filepath.parent + filename = filepath.name + # GIT see also core/views/uploads.py dwgupload() + # GIT see also core/views/expo.py editexpopage() + if encoding: + mode = "w" + kwargs = {"encoding": encoding} + else: + mode = "wb" + kwargs = {} + try: + with open(filepath, mode, **kwargs) as f: + print(f'WRITING{cwd}---{filename} ') + # as the wsgi process www-data, we have group write-access but are not owner, so cannot chmod. + # os.chmod(filepath, 0o664) # set file permissions to rw-rw-r-- + f.write(content) + except PermissionError: + raise WriteAndCommitError(f'CANNOT save this file.\nPERMISSIONS incorrectly set on server for this file {filename}. Ask a nerd to fix this.') + + cp_add = subprocess.run([git, "add", filename], cwd=cwd, capture_output=True, text=True) + if cp_add.returncode != 0: + msgdata = 'Ask a nerd to fix this.\n\n' + cp_add.stderr + '\n\n' + cp_add.stdout + '\n\nreturn code: ' + str(cp_add.returncode) + raise WriteAndCommitError(f'CANNOT git on server for this file {filename}. Edits saved but not added to git.\n\n' + msgdata) + + cp_commit = subprocess.run([git, "commit", "-m", message], cwd=cwd, capture_output=True, text=True) + # This produces return code = 1 if it commits OK, but when the repo still needs to be pushed to origin/expoweb + if cp_commit.returncode != 0 and cp_commit.stdout != 'nothing to commit, working tree clean': + msgdata = 'Ask a nerd to fix this.\n\n' + cp_commit.stderr + '\n\n' + cp_commit.stdout + '\n\nreturn code: ' + str(cp_commit.returncode) + print(msgdata) + raise WriteAndCommitError(f'Error code with git on server for this file {filename}. Edits saved, added to git, but NOT committed.\n\n' + msgdata) + + except subprocess.SubprocessError: + raise WriteAndCommitError(f'CANNOT git on server for this file {filename}. Subprocess error. Edits not saved.\nAsk a nerd to fix this.') + +class WriteAndCommitError(Exception): + """Exception class for errors writing files and comitting them to git""" + def __init__(self, message): + self.message = message + + def __str__(self): + return f'WriteAndCommitError: {self.message}' + def writetrogglefile(filepath, filecontent): - '''Set permissions to rw-rw-r-- and commit the new saved file to git + '''Commit the new saved file to git Callers to cave.writeDataFile() or entrance.writeDataFile() should handle the exception PermissionsError explicitly ''' # GIT see also core/views/expo.py editexpopage() # GIT see also core/views/uploads.py dwgupload() + # Called from core/models/caves.py Cave.writeDataFile() Entrance.writeDataFile() filepath = Path(filepath) cwd = filepath.parent filename = filepath.name diff --git a/core/views/expo.py b/core/views/expo.py index 7b0d283..2a4c1c7 100644 --- a/core/views/expo.py +++ b/core/views/expo.py @@ -21,7 +21,7 @@ from troggle.core.models.caves import Cave import troggle.core.views.caves import troggle.settings as settings -from troggle.lib import version_control +from troggle.core.utils import write_and_commit, WriteAndCommitError from troggle.core.views.editor_helpers import HTMLarea @@ -339,8 +339,8 @@ def editexpopage(request, path): if request.method == 'POST': # If the form has been submitted... pageform = ExpoPageForm(request.POST) # A form bound to the POST data if pageform.is_valid():# Form valid therefore write file - print("### \n", str(pageform)[0:300]) - print("### \n csrfmiddlewaretoken: ",request.POST['csrfmiddlewaretoken']) + #print("### \n", str(pageform)[0:300]) + #print("### \n csrfmiddlewaretoken: ",request.POST['csrfmiddlewaretoken']) if filefound: headmatch = re.match(r"(.*)