zola/components/site/benches/gen.py

"""
Generates test sites for use in benchmark.
Tested with python3 and probably does not work on Windows.
"""
import datetime
import os
import random
import shutil


TAGS = ["a", "b", "c", "d", "e", "f", "g"]
CATEGORIES = ["c1", "c2", "c3", "c4"]

PAGE = """
+++
title = "Hello"
date = REPLACE_DATE

[taxonomies]
tags = REPLACE_TAG
categories = ["REPLACE_CATEGORY"]
+++

# Modus cognitius profanam ne duae virtutis mundi

## Ut vita

Lorem markdownum litora, care ponto nomina, et ut aspicit gelidas sui et
purpureo genuit. Tamen colla venientis [delphina](http://nil-sol.com/ecquis)
Tusci et temptata citaeque curam isto ubi vult vulnere reppulit.

- Seque vidit flendoque de quodam
- Dabit minimos deiecto caputque noctis pluma
- Leti coniunx est Helicen
- Illius pulvereumque Icare inpositos
- Vivunt pereo pluvio tot ramos Olenios gelidis
- Quater teretes natura inde

### A subsection

Protinus dicunt, breve per, et vivacis genus Orphei munere. Me terram [dimittere
casside](http://corpus.org/) pervenit saxo primoque frequentat genuum sorori
praeferre causas Libys. Illud in serpit adsuetam utrimque nunc haberent,
**terrae si** veni! Hectoreis potes sumite [Mavortis retusa](http://tua.org/)
granum captantur potuisse Minervae, frugum.

> Clivo sub inprovisoque nostrum minus fama est, discordia patrem petebat precatur
absumitur, poena per sit. Foramina *tamen cupidine* memor supplex tollentes
dictum unam orbem, Anubis caecae. Viderat formosior tegebat satis, Aethiopasque
sit submisso coniuge tristis ubi!

## Praeceps Corinthus totidem quem crus vultum cape

```rs
#[derive(Debug)]
pub struct Site {
    /// The base path of the zola site
    pub base_path: PathBuf,
    /// The parsed config for the site
    pub config: Config,
    pub pages: HashMap<PathBuf, Page>,
    pub sections: HashMap<PathBuf, Section>,
    pub tera: Tera,
    live_reload: bool,
    output_path: PathBuf,
    static_path: PathBuf,
    pub tags: Option<Taxonomy>,
    pub categories: Option<Taxonomy>,
    /// A map of all .md files (section and pages) and their permalink
    /// We need that if there are relative links in the content that need to be resolved
    pub permalinks: HashMap<String, String>,
}
```

## More stuff
And a shortcode:

{{ youtube(id="my_youtube_id") }}

### Another subsection
Gotta make the toc do a little bit of work

# A big title

- hello
- world
- !

```py
if __name__ == "__main__":
    gen_site("basic-blog", [""], 250, paginate=True)
```
"""


def gen_skeleton(name, is_blog):
    if os.path.exists(name):
        shutil.rmtree(name)

    os.makedirs(os.path.join(name, "content"))
    os.makedirs(os.path.join(name, "static"))

    with open(os.path.join(name, "config.toml"), "w") as f:
        if is_blog:
            f.write("""
title = "My site"
base_url = "https://replace-this-with-your-url.com"
theme = "sample"

taxonomies = [
 {name = "tags", rss = true}, 
 {name = "categories"}
]

[extra.author]
name = "Vincent Prouillet"
""")
        else:
            f.write("""
title = "My site"
base_url = "https://replace-this-with-your-url.com"
theme = "sample"

[extra.author]
name = "Vincent Prouillet"
""")

    # Re-use the test templates
    shutil.copytree("../../../test_site/templates", os.path.join(name, "templates"))
    shutil.copytree("../../../test_site/themes", os.path.join(name, "themes"))


def gen_section(path, num_pages, is_blog):
    with open(os.path.join(path, "_index.md"), "w") as f:
        if is_blog:
            f.write("""
+++
paginate_by = 5
sort_by = "date"
template = "section_paginated.html"
+++
""")
        else:
            f.write("+++\n+++\n")

    day = datetime.date.today()
    for (i, page) in enumerate(range(0, num_pages)):
        with open(os.path.join(path, "page-{}.md".format(i)), "w") as f:
            f.write(
                PAGE
                .replace("REPLACE_DATE", str(day + datetime.timedelta(days=1)))
                .replace("REPLACE_CATEGORY", random.choice(CATEGORIES))
                .replace("REPLACE_TAG", str([random.choice(TAGS), random.choice(TAGS)]))
            )


def gen_site(name, sections, num_pages_per_section, is_blog=False):
    gen_skeleton(name, is_blog)

    for section in sections:
        path = os.path.join(name, "content", section) if section else os.path.join(name, "content")
        if section:
            os.makedirs(path)
        gen_section(path, num_pages_per_section, is_blog)


if __name__ == "__main__":
    gen_site("small-blog", [""], 30, is_blog=True)
    gen_site("medium-blog", [""], 250, is_blog=True)
    gen_site("big-blog", [""], 1000, is_blog=True)
    gen_site("huge-blog", [""], 10000, is_blog=True)
    gen_site("extra-huge-blog", [""], 100000, is_blog=True)

    gen_site("small-kb", ["help", "help1", "help2", "help3", "help4", "help5", "help6", "help7", "help8", "help9"], 10)
    gen_site("medium-kb", ["help", "help1", "help2", "help3", "help4", "help5", "help6", "help7", "help8", "help9"], 100)
    gen_site("huge-kb", ["help", "help1", "help2", "help3", "help4", "help5", "help6", "help7", "help8", "help9"], 1000)