mirror of
https://github.com/satwikkansal/wtfpython
synced 2024-11-22 19:14:24 +01:00
Add json generator
This commit is contained in:
parent
ac379c9c03
commit
596186d81c
156
irrelevant/json_generator.py
Normal file
156
irrelevant/json_generator.py
Normal file
@ -0,0 +1,156 @@
|
|||||||
|
import pprint
|
||||||
|
fname = "/Users/300041709/code/self/wtfpython/README.md"
|
||||||
|
examples = []
|
||||||
|
|
||||||
|
# The globals
|
||||||
|
current_example = 1
|
||||||
|
current_section_name = ""
|
||||||
|
|
||||||
|
|
||||||
|
def parse_example_parts(lines):
|
||||||
|
parts = {
|
||||||
|
"build_up": [],
|
||||||
|
"explanation": []
|
||||||
|
}
|
||||||
|
next_line = next(lines)
|
||||||
|
sequence_num = 1
|
||||||
|
content = []
|
||||||
|
# store build_up till an H4 (explanation) is encountered
|
||||||
|
while not next_line.startswith("#### "):
|
||||||
|
# Watching out for the snippets
|
||||||
|
if next_line.startswith("```"):
|
||||||
|
# It's a snippet, whatever found until now is text
|
||||||
|
if content:
|
||||||
|
parts["build_up"].append(
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
sequence_num += 1
|
||||||
|
content = []
|
||||||
|
|
||||||
|
next_line = next(lines)
|
||||||
|
while not next_line.startswith("```"):
|
||||||
|
content.append(next_line)
|
||||||
|
next_line = next(lines)
|
||||||
|
# Snippet is over
|
||||||
|
parts["build_up"].append(
|
||||||
|
{
|
||||||
|
"type": "code",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
sequence_num += 1
|
||||||
|
content = []
|
||||||
|
next_line = next(lines)
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
# It's a text, go on.
|
||||||
|
content.append(next_line)
|
||||||
|
next_line = next(lines)
|
||||||
|
|
||||||
|
# Explanation encountered, save any content till now (if any)
|
||||||
|
if content:
|
||||||
|
parts["build_up"].append(
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Reset stuff
|
||||||
|
sequence_num = 1
|
||||||
|
content = []
|
||||||
|
|
||||||
|
# store lines again until --- or another H3 is encountered
|
||||||
|
while not (next_line.startswith("---") or
|
||||||
|
next_line.startswith("### ")):
|
||||||
|
|
||||||
|
if next_line.startswith("```"):
|
||||||
|
# It's a snippet, whatever found until now is text
|
||||||
|
if content:
|
||||||
|
parts["explanation"].append(
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
sequence_num += 1
|
||||||
|
content = []
|
||||||
|
|
||||||
|
next_line = next(lines)
|
||||||
|
while not next_line.startswith("```"):
|
||||||
|
content.append(next_line)
|
||||||
|
next_line = next(lines)
|
||||||
|
# Snippet is over
|
||||||
|
parts["explanation"].append(
|
||||||
|
{
|
||||||
|
"type": "code",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
sequence_num += 1
|
||||||
|
content = []
|
||||||
|
next_line = next(lines)
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
# It's a text, go on.
|
||||||
|
content.append(next_line)
|
||||||
|
next_line = next(lines)
|
||||||
|
|
||||||
|
# All done
|
||||||
|
if content:
|
||||||
|
parts["explanation"].append(
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"sequence_num": sequence_num,
|
||||||
|
"value": content
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
return next_line, parts
|
||||||
|
|
||||||
|
|
||||||
|
with open(fname, 'r+', encoding="utf-8") as f:
|
||||||
|
lines = iter(f.readlines())
|
||||||
|
line = next(lines)
|
||||||
|
result = []
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
if line.startswith("## "):
|
||||||
|
# A section is encountered
|
||||||
|
current_section_name = line.replace("## ", "").strip()
|
||||||
|
line = next(lines)
|
||||||
|
# Until a new section is encountered
|
||||||
|
while not (line.startswith("## " )):
|
||||||
|
# check if it's a H3
|
||||||
|
if line.startswith("### "):
|
||||||
|
# An example is encountered
|
||||||
|
title = line.replace("### ", "")
|
||||||
|
example_details = {
|
||||||
|
"id": current_example,
|
||||||
|
"title": line.replace("### ", ""),
|
||||||
|
"section": current_section_name
|
||||||
|
}
|
||||||
|
line, example_details["parts"] = parse_example_parts(lines)
|
||||||
|
result.append(example_details)
|
||||||
|
current_example += 1
|
||||||
|
else:
|
||||||
|
# todo catch section text
|
||||||
|
line = next(lines)
|
||||||
|
else:
|
||||||
|
line = next(lines)
|
||||||
|
|
||||||
|
except StopIteration:
|
||||||
|
pprint.pprint(result, indent=2)
|
||||||
|
print(len(result))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user