szuru2shimmie2/finalizer.nim

36 lines
1.0 KiB
Nim

import std/parsecsv
import std/strutils
import std/osproc
var safety: CsvParser
###Translate Them###
safety.open("post_safety.csv")
var f = open("post_ids_with_named_tags.csv", fmAppend)
f.writeLine("post_id,tag_name")
safety.readHeaderRow()
while safety.readRow():
var command = "find images/ -iname " & safety.rowEntry("id") & "\\.*"
var filename = execCmdEx(command).output.replace("\n")
var tags: string
var rating: char
if len(readFile("PostsWithTags/" & safety.rowEntry("id"))) > 0:
tags = readFile("PostsWithTags/" & safety.rowEntry("id"))
tags = tags[1..len(tags)-1]
else:
tags = "tagme"
if safety.rowEntry("safety") == "safe":
rating = 's'
elif safety.rowEntry("safety") == "questionable":
rating = 'q'
elif safety.rowEntry("safety") == "explicit":
rating = 'e'
let finalFile = open("finalized.csv", fmAppend)
finalFile.writeLine "\"" & filename & "\"" & ",\"" & tags & "\",\"\",\"" & rating & "\",\"thumbnail" & filename & "\""
finalFile.close()