mirror of
https://github.com/satwikkansal/wtfpython
synced 2024-11-22 02:54:25 +01:00
Add json generator
This commit is contained in:
parent
03719dea45
commit
290921fe12
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