mirror of https://gitlab.com/octtspacc/staticoso
143 lines
3.3 KiB
Python
143 lines
3.3 KiB
Python
""" ================================= |
|
|
| This file is part of |
|
|
| staticoso |
|
|
| Just a simple Static Site Generator |
|
|
| |
|
|
| Licensed under the AGPLv3 license |
|
|
| Copyright (C) 2022, OctoSpacc |
|
|
| ================================= """
|
|
|
|
import json
|
|
import os
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
FileExtensions = {
|
|
'Pages': ('htm', 'html', 'md', 'pug', 'txt'),
|
|
'Tmp': ('md', 'pug', 'txt')}
|
|
|
|
def ReadFile(p):
|
|
try:
|
|
with open(p, 'r') as f:
|
|
return f.read()
|
|
except Exception:
|
|
print("[E] Error reading file {}".format(p))
|
|
return None
|
|
|
|
def WriteFile(p, c):
|
|
try:
|
|
with open(p, 'w') as f:
|
|
f.write(c)
|
|
return True
|
|
except Exception:
|
|
print("[E] Error writing file {}".format(p))
|
|
return False
|
|
|
|
def FileToStr(File, Truncate=''):
|
|
return str(File)[len(Truncate):]
|
|
|
|
# https://stackoverflow.com/a/15664273
|
|
def IgnoreFiles(Dir, Files):
|
|
return [f for f in Files if os.path.isfile(os.path.join(Dir, f))]
|
|
|
|
def LoadFromDir(Dir, Rglob):
|
|
Contents = {}
|
|
for File in Path(Dir).rglob(Rglob):
|
|
File = str(File)[len(Dir)+1:]
|
|
Contents.update({File: ReadFile('{}/{}'.format(Dir, File))})
|
|
return Contents
|
|
|
|
def StripExt(Path):
|
|
return ".".join(Path.split('.')[:-1])
|
|
|
|
def UndupeStr(Str, Known, Split):
|
|
while Str in Known:
|
|
Sections = Str.split(Split)
|
|
try:
|
|
Sections[-1] = str(int(Sections[-1]) + 1)
|
|
except ValueError:
|
|
Sections[-1] = Sections[-1] + str(Split) + '2'
|
|
Str = Split.join(Sections)
|
|
return Str
|
|
|
|
def DashifyStr(s, Limit=32):
|
|
Str, lc = '', Limit
|
|
for c in s[:Limit].replace(' ','-').replace(' ','-'):
|
|
if c.lower() in '0123456789qwfpbjluyarstgmneiozxcdvkh-':
|
|
Str += c
|
|
return '-' + Str
|
|
|
|
def GetPathLevels(Path, AsNum=False, Add=0, Sub=0):
|
|
n = Path.count('/') + Add - Sub
|
|
return n if AsNum else '../' * n
|
|
|
|
# https://stackoverflow.com/a/34445090
|
|
def FindAllIndex(Str, Sub):
|
|
i = Str.find(Sub)
|
|
while i != -1:
|
|
yield i
|
|
i = Str.find(Sub, i+1)
|
|
|
|
def ReplWithEsc(Str, Find, Repl, Esc='\\'):
|
|
New = ''
|
|
Sects = Str.split(Find)
|
|
for i,e in enumerate(Sects):
|
|
if i == 0:
|
|
New += e
|
|
elif i > 0:
|
|
if Sects[i-1].endswith(Esc*2):
|
|
New = New[:-1]
|
|
New += Repl + e
|
|
elif Sects[i-1].endswith(Esc):
|
|
New += Find + e
|
|
else:
|
|
New += Repl + e
|
|
return New
|
|
|
|
def NumsFromFileName(Path):
|
|
Name = Path.split('/')[-1]
|
|
Split = len(Name)
|
|
for i,e in enumerate(Name):
|
|
if e.lower() in 'qwfpbjluyarstgmneiozxcdvkh':
|
|
return Name[:i]
|
|
return Path
|
|
|
|
def RevSort(List):
|
|
List.sort()
|
|
List.reverse()
|
|
return List
|
|
|
|
def FileNameDateSort(Old): # TODO: Test this for files not starting with date, and dated folders
|
|
New = []
|
|
Old.sort()
|
|
New.insert(0, Old[0])
|
|
for i,e in enumerate(Old):
|
|
if i == 0:
|
|
continue
|
|
Done = False
|
|
for j,f in enumerate(New):
|
|
if NumsFromFileName(e) != e and NumsFromFileName(f) != f and NumsFromFileName(e) < NumsFromFileName(f):
|
|
New.insert(j, e)
|
|
Done = True
|
|
break
|
|
if not Done:
|
|
New += [e]
|
|
return New
|
|
|
|
def FirstRealItem(List):
|
|
return next(e for e in List if e)
|
|
|
|
def GetFullDate(Date):
|
|
if not Date:
|
|
return None
|
|
return datetime.strftime(datetime.strptime(Date, '%Y-%m-%d'), '%Y-%m-%dT%H:%M+00:00')
|
|
|
|
def LoadLocale(Lang):
|
|
Lang = Lang + '.json'
|
|
Folder = os.path.dirname(os.path.abspath(__file__)) + '/../../Locale/'
|
|
File = ReadFile(Folder + Lang)
|
|
if File:
|
|
return json.loads(File)
|
|
else:
|
|
return json.loads(ReadFile(Folder + 'en.json'))
|