Inprogress refactor trying f-strings as template language

2020-04-25 16:09:11 +01:00
parent 0224d72931
commit 4f75eaabcd
4 changed files with 91 additions and 13 deletions
@@ -0,0 +1,18 @@
+import os
+import io
+from typing import List, Tuple
+
+
+class TemplateProject:
+    files: List[Tuple[str, str]]
+
+    def __init__(self, files):
+        self.files = files
+
+    def generate(self, path):
+        for filename, content in self.files:
+            file_path = os.path.join(path, filename)
+            os.makedirs(os.path.dirname(file_path), exist_ok=True)
+
+            with io.open(file_path, 'w', encoding='utf8') as f:
+                f.write(content)
@@ -0,0 +1,60 @@
+import textwrap
+
+def indent(str, level):
+    return str
+
+def CTemplateProject(TemplateProject):
+    files = {
+        'spam.c': textwrap.dedent(fr'''
+            #include <Python.h>
+
+            {spam_c_top_level_add}
+
+            static PyObject *
+            spam_system(PyObject *self, PyObject *args)
+            {{
+                const char *command;
+                int sts;
+                if (!PyArg_ParseTuple(args, "s", &command))
+                    return NULL;
+                sts = system(command);
+                {indent(spam_c_function_add, 4)}
+                return PyLong_FromLong(sts);
+            }}
+
+            /* Module initialization */
+
+            #if PY_MAJOR_VERSION >= 3
+                #define MOD_INIT(name) PyMODINIT_FUNC PyInit_##name(void)
+                #define MOD_DEF(m, name, doc, methods, module_state_size) \
+                    static struct PyModuleDef moduledef = { \
+                        PyModuleDef_HEAD_INIT, name, doc, module_state_size, methods, }; \
+                    m = PyModule_Create(&moduledef);
+                #define MOD_RETURN(m) return m;
+            #else
+                #define MOD_INIT(name) PyMODINIT_FUNC init##name(void)
+                #define MOD_DEF(m, name, doc, methods, module_state_size) \
+                    m = Py_InitModule3(name, methods, doc);
+                #define MOD_RETURN(m) return;
+            #endif
+
+            static PyMethodDef module_methods[] = {
+                {"system", (PyCFunction)spam_system, METH_VARARGS,
+                 "Execute a shell command."},
+                {NULL}  /* Sentinel */
+            };
+
+            MOD_INIT(spam)
+            {
+                PyObject* m;
+
+                MOD_DEF(m,
+                        "spam",
+                        "Example module",
+                        module_methods,
+                        -1)
+
+                MOD_RETURN(m)
+            }
+            ''')
+    }