From f73e2fc527450f713b78154a08b113656a01a9f4 Mon Sep 17 00:00:00 2001 From: Andrii Abramov Date: Thu, 14 Jan 2021 19:05:29 +0200 Subject: [PATCH] Add ScalaGenerator --- .../generators/ScalaGenerator.py | 26 +++++++++++++++++++ .../generators/__init__.py | 2 ++ 2 files changed, 28 insertions(+) create mode 100644 git_contributions_importer/generators/ScalaGenerator.py diff --git a/git_contributions_importer/generators/ScalaGenerator.py b/git_contributions_importer/generators/ScalaGenerator.py new file mode 100644 index 0000000..d1f58a6 --- /dev/null +++ b/git_contributions_importer/generators/ScalaGenerator.py @@ -0,0 +1,26 @@ +#!/usr/bin/python3 + +from . import Generator + + +class ScalaGenerator(Generator): + # one day this will generate awesome random code + + min_content_size = 4 + + def __init__(self): + pass + + def insert(self, content, num): + if len(content) <= self.min_content_size: + content.clear() + content.append('object Main' + self.random_string(5) + ' {') + content.append(' def main(args: Array[String]): Unit = {') + content.append(' }') + content.append('}') + for i in range(num): + content.insert(-2, ' println("' + self.random_string(5) + '");') + + def delete(self, content, num): + for i in range(min(num, len(content) - self.min_content_size)): + content.pop(-3) diff --git a/git_contributions_importer/generators/__init__.py b/git_contributions_importer/generators/__init__.py index 6cbdf10..4cca874 100644 --- a/git_contributions_importer/generators/__init__.py +++ b/git_contributions_importer/generators/__init__.py @@ -3,6 +3,7 @@ from .Generator import Generator from .JsGenerator import JsGenerator from .JavaGenerator import JavaGenerator +from .ScalaGenerator import ScalaGenerator from .CssGenerator import CssGenerator from .CppGenerator import CppGenerator from .CGenerator import CGenerator @@ -20,6 +21,7 @@ '.tex': Generator, '.js': JsGenerator, '.java': JavaGenerator, + '.scala': ScalaGenerator, '.css': CssGenerator, '.scss': CssGenerator, '.cpp': CppGenerator,