|
|
|
@ -3,14 +3,6 @@ import json, macros, options, sequtils, strutils, times, timeutils, unicode,
|
|
|
|
|
|
|
|
|
|
import nre except toSeq
|
|
|
|
|
|
|
|
|
|
const UNDERSCORE_RUNE = "_".toRunes[0]
|
|
|
|
|
const PG_TIMESTAMP_FORMATS = [
|
|
|
|
|
"yyyy-MM-dd HH:mm:sszz",
|
|
|
|
|
"yyyy-MM-dd HH:mm:ss'.'fffzz"
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
var PG_PARTIAL_FORMAT_REGEX = re"(\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.)(\d{1,3})(\S+)?"
|
|
|
|
|
|
|
|
|
|
type
|
|
|
|
|
MutateClauses* = object
|
|
|
|
|
columns*: seq[string]
|
|
|
|
@ -30,7 +22,9 @@ macro modelName*(model: object): string =
|
|
|
|
|
macro modelName*(modelType: type): string =
|
|
|
|
|
return newStrLitNode($modelType.getType[1])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
proc identNameToDb*(name: string): string =
|
|
|
|
|
const UNDERSCORE_RUNE = "_".toRunes[0]
|
|
|
|
|
let nameInRunes = name.toRunes
|
|
|
|
|
var prev: Rune
|
|
|
|
|
var resultRunes = newSeq[Rune]()
|
|
|
|
@ -70,27 +64,41 @@ type DbArrayParseState = enum
|
|
|
|
|
expectStart, inQuote, inVal, expectEnd
|
|
|
|
|
|
|
|
|
|
proc parsePGDatetime*(val: string): DateTime =
|
|
|
|
|
var errStr = ""
|
|
|
|
|
|
|
|
|
|
# Try to parse directly using known format strings.
|
|
|
|
|
for df in PG_TIMESTAMP_FORMATS:
|
|
|
|
|
try: return val.parse(df)
|
|
|
|
|
except: errStr &= "\n\t" & getCurrentExceptionMsg()
|
|
|
|
|
const PG_TIMESTAMP_FORMATS = [
|
|
|
|
|
"yyyy-MM-dd HH:mm:ss",
|
|
|
|
|
"yyyy-MM-dd'T'HH:mm:ss",
|
|
|
|
|
"yyyy-MM-dd HH:mm:sszz",
|
|
|
|
|
"yyyy-MM-dd'T'HH:mm:sszz",
|
|
|
|
|
"yyyy-MM-dd HH:mm:ss'.'fff",
|
|
|
|
|
"yyyy-MM-dd'T'HH:mm:ss'.'fff",
|
|
|
|
|
"yyyy-MM-dd HH:mm:ss'.'fffzz",
|
|
|
|
|
"yyyy-MM-dd'T'HH:mm:ss'.'fffzz",
|
|
|
|
|
"yyyy-MM-dd HH:mm:ss'.'fffzzz",
|
|
|
|
|
"yyyy-MM-dd'T'HH:mm:ss'.'fffzzz",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
var correctedVal = val;
|
|
|
|
|
|
|
|
|
|
# PostgreSQL will truncate any trailing 0's in the millisecond value leading
|
|
|
|
|
# to values like `2020-01-01 16:42.3+00`. This cannot currently be parsed by
|
|
|
|
|
# the standard times format as it expects exactly three digits for
|
|
|
|
|
# millisecond values. So we have to detect this and pad out the millisecond
|
|
|
|
|
# value to 3 digits.
|
|
|
|
|
let PG_PARTIAL_FORMAT_REGEX = re"(\d{4}-\d{2}-\d{2}( |'T')\d{2}:\d{2}:\d{2}\.)(\d{1,2})(\S+)?"
|
|
|
|
|
let match = val.match(PG_PARTIAL_FORMAT_REGEX)
|
|
|
|
|
|
|
|
|
|
if match.isSome:
|
|
|
|
|
let c = match.get.captures
|
|
|
|
|
try:
|
|
|
|
|
let corrected = c[0] & alignLeft(c[1], 3, '0') & c[2]
|
|
|
|
|
return corrected.parse(PG_TIMESTAMP_FORMATS[1])
|
|
|
|
|
except:
|
|
|
|
|
errStr &= "\n\t" & PG_TIMESTAMP_FORMATS[1] &
|
|
|
|
|
" after padding out milliseconds to full 3-digits"
|
|
|
|
|
if c.toSeq.len == 2: correctedVal = c[0] & alignLeft(c[2], 3, '0')
|
|
|
|
|
else: correctedVal = c[0] & alignLeft(c[2], 3, '0') & c[3]
|
|
|
|
|
|
|
|
|
|
var errStr = ""
|
|
|
|
|
|
|
|
|
|
# Try to parse directly using known format strings.
|
|
|
|
|
for df in PG_TIMESTAMP_FORMATS:
|
|
|
|
|
try: return correctedVal.parse(df)
|
|
|
|
|
except: errStr &= "\n\t" & getCurrentExceptionMsg()
|
|
|
|
|
|
|
|
|
|
raise newException(ValueError, "Cannot parse PG date. Tried:" & errStr)
|
|
|
|
|
|
|
|
|
@ -197,6 +205,9 @@ proc createParseStmt*(t, value: NimNode): NimNode =
|
|
|
|
|
elif t.typeKind == ntyInt:
|
|
|
|
|
result = quote do: parseInt(`value`)
|
|
|
|
|
|
|
|
|
|
elif t.typeKind == ntyFloat:
|
|
|
|
|
result = quote do: parseFloat(`value`)
|
|
|
|
|
|
|
|
|
|
elif t.typeKind == ntyBool:
|
|
|
|
|
result = quote do: "true".startsWith(`value`.toLower)
|
|
|
|
|
|
|
|
|
@ -275,7 +286,7 @@ proc typeOfColumn*(modelType: NimNode, colName: string): NimNode =
|
|
|
|
|
|
|
|
|
|
proc isEmpty(val: int): bool = return val == 0
|
|
|
|
|
proc isEmpty(val: UUID): bool = return val.isZero
|
|
|
|
|
proc isEmpty(val: string): bool = return val.isNilOrWhitespace
|
|
|
|
|
proc isEmpty(val: string): bool = return val.isEmptyOrWhitespace
|
|
|
|
|
|
|
|
|
|
macro populateMutateClauses*(t: typed, newRecord: bool, mc: var MutateClauses): untyped =
|
|
|
|
|
|
|
|
|
|