File write style

main
Bob Mottram 2020-07-12 20:04:58 +00:00
parent e1347efb20
commit fe5e21ee7d
16 changed files with 47 additions and 41 deletions

View File

@ -67,7 +67,7 @@ def removeGlobalBlock(baseDir: str,
if os.path.isfile(unblockingFilename): if os.path.isfile(unblockingFilename):
if unblockHandle in open(unblockingFilename).read(): if unblockHandle in open(unblockingFilename).read():
with open(unblockingFilename, 'r') as fp: with open(unblockingFilename, 'r') as fp:
with open(unblockingFilename + '.new', 'w') as fpnew: with open(unblockingFilename + '.new', 'w+') as fpnew:
for line in fp: for line in fp:
handle = line.replace('\n', '').replace('\r', '') handle = line.replace('\n', '').replace('\r', '')
if unblockHandle not in line: if unblockHandle not in line:
@ -80,7 +80,7 @@ def removeGlobalBlock(baseDir: str,
if os.path.isfile(unblockingFilename): if os.path.isfile(unblockingFilename):
if unblockHashtag + '\n' in open(unblockingFilename).read(): if unblockHashtag + '\n' in open(unblockingFilename).read():
with open(unblockingFilename, 'r') as fp: with open(unblockingFilename, 'r') as fp:
with open(unblockingFilename + '.new', 'w') as fpnew: with open(unblockingFilename + '.new', 'w+') as fpnew:
for line in fp: for line in fp:
blockLine = \ blockLine = \
line.replace('\n', '').replace('\r', '') line.replace('\n', '').replace('\r', '')
@ -104,7 +104,7 @@ def removeBlock(baseDir: str, nickname: str, domain: str,
if os.path.isfile(unblockingFilename): if os.path.isfile(unblockingFilename):
if unblockHandle in open(unblockingFilename).read(): if unblockHandle in open(unblockingFilename).read():
with open(unblockingFilename, 'r') as fp: with open(unblockingFilename, 'r') as fp:
with open(unblockingFilename + '.new', 'w') as fpnew: with open(unblockingFilename + '.new', 'w+') as fpnew:
for line in fp: for line in fp:
handle = line.replace('\n', '').replace('\r', '') handle = line.replace('\n', '').replace('\r', '')
if unblockHandle not in line: if unblockHandle not in line:

View File

@ -57,7 +57,7 @@ def undoBookmarksCollectionEntry(recentPostsCache: {},
indexStr = '' indexStr = ''
with open(bookmarksIndexFilename, 'r') as indexFile: with open(bookmarksIndexFilename, 'r') as indexFile:
indexStr = indexFile.read().replace(bookmarkIndex + '\n', '') indexStr = indexFile.read().replace(bookmarkIndex + '\n', '')
bookmarksIndexFile = open(bookmarksIndexFilename, 'w') bookmarksIndexFile = open(bookmarksIndexFilename, 'w+')
if bookmarksIndexFile: if bookmarksIndexFile:
bookmarksIndexFile.write(indexStr) bookmarksIndexFile.write(indexStr)
bookmarksIndexFile.close() bookmarksIndexFile.close()
@ -213,7 +213,7 @@ def updateBookmarksCollection(recentPostsCache: {},
print('WARN: Failed to write entry to bookmarks index ' + print('WARN: Failed to write entry to bookmarks index ' +
bookmarksIndexFilename + ' ' + str(e)) bookmarksIndexFilename + ' ' + str(e))
else: else:
bookmarksIndexFile = open(bookmarksIndexFilename, 'w') bookmarksIndexFile = open(bookmarksIndexFilename, 'w+')
if bookmarksIndexFile: if bookmarksIndexFile:
bookmarksIndexFile.write(bookmarkIndex + '\n') bookmarksIndexFile.write(bookmarkIndex + '\n')
bookmarksIndexFile.close() bookmarksIndexFile.close()

View File

@ -292,7 +292,7 @@ class PubServer(BaseHTTPRequestHandler):
if not minimal and minimalFileExists: if not minimal and minimalFileExists:
os.remove(minimalFilename) os.remove(minimalFilename)
elif minimal and not minimalFileExists: elif minimal and not minimalFileExists:
with open(minimalFilename, 'w') as fp: with open(minimalFilename, 'w+') as fp:
fp.write('\n') fp.write('\n')
def _sendReplyToQuestion(self, nickname: str, messageId: str, def _sendReplyToQuestion(self, nickname: str, messageId: str,
@ -539,7 +539,7 @@ class PubServer(BaseHTTPRequestHandler):
if not etag: if not etag:
etag = sha1(data).hexdigest() # nosec etag = sha1(data).hexdigest() # nosec
try: try:
with open(mediaFilename + '.etag', 'w') as etagFile: with open(mediaFilename + '.etag', 'w+') as etagFile:
etagFile.write(etag) etagFile.write(etag)
except BaseException: except BaseException:
pass pass
@ -5108,7 +5108,7 @@ class PubServer(BaseHTTPRequestHandler):
mediaBinary = avFile.read() mediaBinary = avFile.read()
etag = sha1(mediaBinary).hexdigest() # nosec etag = sha1(mediaBinary).hexdigest() # nosec
try: try:
with open(mediaTagFilename, 'w') as etagFile: with open(mediaTagFilename, 'w+') as etagFile:
etagFile.write(etag) etagFile.write(etag)
except BaseException: except BaseException:
pass pass
@ -5255,7 +5255,7 @@ class PubServer(BaseHTTPRequestHandler):
self.server.baseDir + '/accounts/' + \ self.server.baseDir + '/accounts/' + \
nickname + '@' + self.server.domain + '/.lastUsed' nickname + '@' + self.server.domain + '/.lastUsed'
try: try:
lastUsedFile = open(lastUsedFilename, 'w') lastUsedFile = open(lastUsedFilename, 'w+')
if lastUsedFile: if lastUsedFile:
lastUsedFile.write(str(int(time.time()))) lastUsedFile.write(str(int(time.time())))
lastUsedFile.close() lastUsedFile.close()
@ -5903,7 +5903,7 @@ class PubServer(BaseHTTPRequestHandler):
loginNickname + ' ' + str(e)) loginNickname + ' ' + str(e))
else: else:
try: try:
with open(saltFilename, 'w') as fp: with open(saltFilename, 'w+') as fp:
fp.write(salt) fp.write(salt)
except Exception as e: except Exception as e:
print('WARN: Unable to save salt for ' + print('WARN: Unable to save salt for ' +
@ -5917,7 +5917,7 @@ class PubServer(BaseHTTPRequestHandler):
self.server.baseDir+'/accounts/' + \ self.server.baseDir+'/accounts/' + \
loginHandle + '/.token' loginHandle + '/.token'
try: try:
with open(tokenFilename, 'w') as fp: with open(tokenFilename, 'w+') as fp:
fp.write(token) fp.write(token)
except Exception as e: except Exception as e:
print('WARN: Unable to save token for ' + print('WARN: Unable to save token for ' +

View File

@ -479,7 +479,7 @@ if args.socnet:
httpPrefix, debug, httpPrefix, debug,
__version__) __version__)
try: try:
with open('socnet.dot', 'w') as fp: with open('socnet.dot', 'w+') as fp:
fp.write(dotGraph) fp.write(dotGraph)
print('Saved to socnet.dot') print('Saved to socnet.dot')
except BaseException: except BaseException:

View File

@ -32,7 +32,7 @@ def removeFilter(baseDir: str, nickname: str, domain: str,
if os.path.isfile(filtersFilename): if os.path.isfile(filtersFilename):
if words in open(filtersFilename).read(): if words in open(filtersFilename).read():
with open(filtersFilename, 'r') as fp: with open(filtersFilename, 'r') as fp:
with open(filtersFilename + '.new', 'w') as fpnew: with open(filtersFilename + '.new', 'w+') as fpnew:
for line in fp: for line in fp:
line = line.replace('\n', '') line = line.replace('\n', '')
if line != words: if line != words:

View File

@ -29,7 +29,7 @@ def receivingCalendarEvents(baseDir: str, nickname: str, domain: str,
# create a new calendar file from the following file # create a new calendar file from the following file
with open(followingFilename, 'r') as followingFile: with open(followingFilename, 'r') as followingFile:
followingHandles = followingFile.read() followingHandles = followingFile.read()
with open(calendarFilename, 'w') as fp: with open(calendarFilename, 'w+') as fp:
fp.write(followingHandles) fp.write(followingHandles)
return handle + '\n' in open(calendarFilename).read() return handle + '\n' in open(calendarFilename).read()
@ -65,7 +65,7 @@ def receiveCalendarEvents(baseDir: str, nickname: str, domain: str,
# create a new calendar file from the following file # create a new calendar file from the following file
with open(followingFilename, 'r') as followingFile: with open(followingFilename, 'r') as followingFile:
followingHandles = followingFile.read() followingHandles = followingFile.read()
with open(calendarFilename, 'w') as fp: with open(calendarFilename, 'w+') as fp:
fp.write(followingHandles) fp.write(followingHandles)
# already in the calendar file? # already in the calendar file?
@ -75,14 +75,14 @@ def receiveCalendarEvents(baseDir: str, nickname: str, domain: str,
return return
# remove from calendar file # remove from calendar file
followingHandles = followingHandles.replace(handle + '\n', '') followingHandles = followingHandles.replace(handle + '\n', '')
with open(calendarFilename, 'w') as fp: with open(calendarFilename, 'w+') as fp:
fp.write(followingHandles) fp.write(followingHandles)
else: else:
# not already in the calendar file # not already in the calendar file
if add: if add:
# append to the list of handles # append to the list of handles
followingHandles += handle + '\n' followingHandles += handle + '\n'
with open(calendarFilename, 'w') as fp: with open(calendarFilename, 'w+') as fp:
fp.write(followingHandles) fp.write(followingHandles)

View File

@ -1702,7 +1702,7 @@ def dmNotify(baseDir: str, handle: str, url: str) -> None:
return return
dmFile = accountDir + '/.newDM' dmFile = accountDir + '/.newDM'
if not os.path.isfile(dmFile): if not os.path.isfile(dmFile):
with open(dmFile, 'w') as fp: with open(dmFile, 'w+') as fp:
fp.write(url) fp.write(url)
@ -1748,10 +1748,16 @@ def likeNotify(baseDir: str, domain: str, onionDomain: str,
prevLikeStr = likeFile.read() prevLikeStr = likeFile.read()
if prevLikeStr == likeStr: if prevLikeStr == likeStr:
return return
with open(prevLikeFile, 'w') as fp: try:
with open(prevLikeFile, 'w+') as fp:
fp.write(likeStr) fp.write(likeStr)
with open(likeFile, 'w') as fp: except BaseException:
pass
try:
with open(likeFile, 'w+') as fp:
fp.write(likeStr) fp.write(likeStr)
except BaseException:
pass
def replyNotify(baseDir: str, handle: str, url: str) -> None: def replyNotify(baseDir: str, handle: str, url: str) -> None:
@ -1762,7 +1768,7 @@ def replyNotify(baseDir: str, handle: str, url: str) -> None:
return return
replyFile = accountDir + '/.newReply' replyFile = accountDir + '/.newReply'
if not os.path.isfile(replyFile): if not os.path.isfile(replyFile):
with open(replyFile, 'w') as fp: with open(replyFile, 'w+') as fp:
fp.write(url) fp.write(url)
@ -1777,7 +1783,7 @@ def gitPatchNotify(baseDir: str, handle: str,
patchFile = accountDir + '/.newPatch' patchFile = accountDir + '/.newPatch'
subject = subject.replace('[PATCH]', '').strip() subject = subject.replace('[PATCH]', '').strip()
handle = '@' + fromNickname + '@' + fromDomain handle = '@' + fromNickname + '@' + fromDomain
with open(patchFile, 'w') as fp: with open(patchFile, 'w+') as fp:
fp.write('git ' + handle + ' ' + subject) fp.write('git ' + handle + ' ' + subject)
@ -1949,7 +1955,7 @@ def inboxUpdateCalendar(baseDir: str, handle: str, postJsonObject: {}) -> None:
calendarNotificationFilename = \ calendarNotificationFilename = \
baseDir + '/accounts/' + handle + '/.newCalendar' baseDir + '/accounts/' + handle + '/.newCalendar'
calendarNotificationFile = \ calendarNotificationFile = \
open(calendarNotificationFilename, 'w') open(calendarNotificationFilename, 'w+')
if calendarNotificationFile: if calendarNotificationFile:
calendarNotificationFile.write('/calendar?year=' + calendarNotificationFile.write('/calendar?year=' +
str(eventYear) + str(eventYear) +

View File

@ -122,7 +122,7 @@ def updateEtag(mediaFilename: str) -> None:
etag = sha1(data).hexdigest() # nosec etag = sha1(data).hexdigest() # nosec
# save the hash # save the hash
try: try:
with open(mediaFilename + '.etag', 'w') as etagFile: with open(mediaFilename + '.etag', 'w+') as etagFile:
etagFile.write(etag) etagFile.write(etag)
except BaseException: except BaseException:
pass pass

View File

@ -25,7 +25,7 @@ def migrateFollows(followFilename: str, oldHandle: str,
newFollowData = followData.replace(oldHandle, newHandle) newFollowData = followData.replace(oldHandle, newHandle)
if followData == newFollowData: if followData == newFollowData:
return return
with open(followFilename, 'w') as followFile: with open(followFilename, 'w+') as followFile:
followFile.write(newFollowData) followFile.write(newFollowData)

View File

@ -1004,7 +1004,7 @@ def isPersonSnoozed(baseDir: str, nickname: str, domain: str,
with open(snoozedFilename, 'r') as snoozedFile: with open(snoozedFilename, 'r') as snoozedFile:
content = snoozedFile.read().replace(replaceStr, '') content = snoozedFile.read().replace(replaceStr, '')
if content: if content:
writeSnoozedFile = open(snoozedFilename, 'w') writeSnoozedFile = open(snoozedFilename, 'w+')
if writeSnoozedFile: if writeSnoozedFile:
writeSnoozedFile.write(content) writeSnoozedFile.write(content)
writeSnoozedFile.close() writeSnoozedFile.close()
@ -1057,7 +1057,7 @@ def personUnsnooze(baseDir: str, nickname: str, domain: str,
with open(snoozedFilename, 'r') as snoozedFile: with open(snoozedFilename, 'r') as snoozedFile:
content = snoozedFile.read().replace(replaceStr, '') content = snoozedFile.read().replace(replaceStr, '')
if content: if content:
writeSnoozedFile = open(snoozedFilename, 'w') writeSnoozedFile = open(snoozedFilename, 'w+')
if writeSnoozedFile: if writeSnoozedFile:
writeSnoozedFile.write(content) writeSnoozedFile.write(content)
writeSnoozedFile.close() writeSnoozedFile.close()

View File

@ -40,7 +40,7 @@ def setPetName(baseDir: str, nickname: str, domain: str,
else: else:
newPetnamesStr += entry newPetnamesStr += entry
# save the updated petnames file # save the updated petnames file
with open(petnamesFilename, 'w') as petnamesFile: with open(petnamesFilename, 'w+') as petnamesFile:
petnamesFile.write(newPetnamesStr) petnamesFile.write(newPetnamesStr)
return True return True
# entry does not exist in the petnames file # entry does not exist in the petnames file
@ -49,7 +49,7 @@ def setPetName(baseDir: str, nickname: str, domain: str,
return True return True
# first entry # first entry
with open(petnamesFilename, 'w') as petnamesFile: with open(petnamesFilename, 'w+') as petnamesFile:
petnamesFile.write(entry) petnamesFile.write(entry)
return True return True

View File

@ -599,7 +599,7 @@ def addSchedulePost(baseDir: str, nickname: str, domain: str,
print('WARN: Failed to write entry to scheduled posts index ' + print('WARN: Failed to write entry to scheduled posts index ' +
scheduleIndexFilename + ' ' + str(e)) scheduleIndexFilename + ' ' + str(e))
else: else:
scheduleFile = open(scheduleIndexFilename, 'w') scheduleFile = open(scheduleIndexFilename, 'w+')
if scheduleFile: if scheduleFile:
scheduleFile.write(indexStr + '\n') scheduleFile.write(indexStr + '\n')
scheduleFile.close() scheduleFile.close()
@ -1337,7 +1337,7 @@ def createReportPost(baseDir: str,
if os.path.isfile(newReportFile): if os.path.isfile(newReportFile):
continue continue
try: try:
with open(newReportFile, 'w') as fp: with open(newReportFile, 'w+') as fp:
fp.write(toUrl + '/moderation') fp.write(toUrl + '/moderation')
except BaseException: except BaseException:
pass pass

View File

@ -181,7 +181,7 @@ def addShare(baseDir: str,
if not os.path.isfile(newShareFile): if not os.path.isfile(newShareFile):
nickname = handle.split('@')[0] nickname = handle.split('@')[0]
try: try:
with open(newShareFile, 'w') as fp: with open(newShareFile, 'w+') as fp:
fp.write(httpPrefix + '://' + domainFull + fp.write(httpPrefix + '://' + domainFull +
'/users/' + nickname + '/tlshares') '/users/' + nickname + '/tlshares')
except BaseException: except BaseException:

View File

@ -105,7 +105,7 @@ def setThemeFromDict(baseDir: str, name: str, themeParams: {}) -> None:
for paramName, paramValue in themeParams.items(): for paramName, paramValue in themeParams.items():
css = setCSSparam(css, paramName, paramValue) css = setCSSparam(css, paramName, paramValue)
filename = baseDir + '/' + filename filename = baseDir + '/' + filename
with open(filename, 'w') as cssfile: with open(filename, 'w+') as cssfile:
cssfile.write(css) cssfile.write(css)
@ -124,11 +124,11 @@ def enableGrayscale(baseDir: str) -> None:
css.replace('body, html {', css.replace('body, html {',
'body, html {\n filter: grayscale(100%);') 'body, html {\n filter: grayscale(100%);')
filename = baseDir + '/' + filename filename = baseDir + '/' + filename
with open(filename, 'w') as cssfile: with open(filename, 'w+') as cssfile:
cssfile.write(css) cssfile.write(css)
grayscaleFilename = baseDir + '/accounts/.grayscale' grayscaleFilename = baseDir + '/accounts/.grayscale'
if not os.path.isfile(grayscaleFilename): if not os.path.isfile(grayscaleFilename):
with open(grayscaleFilename, 'w') as grayfile: with open(grayscaleFilename, 'w+') as grayfile:
grayfile.write(' ') grayfile.write(' ')
@ -146,7 +146,7 @@ def disableGrayscale(baseDir: str) -> None:
css = \ css = \
css.replace('\n filter: grayscale(100%);', '') css.replace('\n filter: grayscale(100%);', '')
filename = baseDir + '/' + filename filename = baseDir + '/' + filename
with open(filename, 'w') as cssfile: with open(filename, 'w+') as cssfile:
cssfile.write(css) cssfile.write(css)
grayscaleFilename = baseDir + '/accounts/.grayscale' grayscaleFilename = baseDir + '/accounts/.grayscale'
if os.path.isfile(grayscaleFilename): if os.path.isfile(grayscaleFilename):
@ -187,7 +187,7 @@ def setCustomFont(baseDir: str):
customFontType + "')") customFontType + "')")
css = setCSSparam(css, "*font-family", "'CustomFont'") css = setCSSparam(css, "*font-family", "'CustomFont'")
filename = baseDir + '/' + filename filename = baseDir + '/' + filename
with open(filename, 'w') as cssfile: with open(filename, 'w+') as cssfile:
cssfile.write(css) cssfile.write(css)

View File

@ -51,7 +51,7 @@ def saveJson(jsonObject: {}, filename: str) -> bool:
tries = 0 tries = 0
while tries < 5: while tries < 5:
try: try:
with open(filename, 'w') as fp: with open(filename, 'w+') as fp:
fp.write(json.dumps(jsonObject)) fp.write(json.dumps(jsonObject))
return True return True
except BaseException: except BaseException:

View File

@ -3337,7 +3337,7 @@ def saveIndividualPostAsHtmlToCache(baseDir: str,
os.mkdir(htmlPostCacheDir) os.mkdir(htmlPostCacheDir)
try: try:
with open(cachedPostFilename, 'w') as fp: with open(cachedPostFilename, 'w+') as fp:
fp.write(postHtml) fp.write(postHtml)
return True return True
except Exception as e: except Exception as e: