43 lines
1.2 KiB
Python
Executable file
43 lines
1.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import os
|
|
import json
|
|
import glob
|
|
import inflection
|
|
from datetime import datetime
|
|
from collections import OrderedDict
|
|
|
|
migration_template = """class {} < ActiveRecord::Migration[5.2]
|
|
def self.up
|
|
# add your code here
|
|
end
|
|
|
|
def self.down
|
|
# add your code here
|
|
end
|
|
end
|
|
"""
|
|
|
|
|
|
def load_skeleton():
|
|
t = glob.glob('*.szpm.template')
|
|
if len(t) != 1:
|
|
raise Exception("Cannot find szpm template")
|
|
with open(t[0], 'r', encoding='utf-8') as f:
|
|
skeleton = json.load(f, object_pairs_hook=OrderedDict)
|
|
return skeleton
|
|
|
|
|
|
def main():
|
|
skeleton = load_skeleton()
|
|
name = skeleton["name"].lower()
|
|
raw_name = input("Enter migration name: ")
|
|
migration_base_name = "{}_{}".format(name, inflection.underscore(raw_name))
|
|
migration_name = inflection.camelize(migration_base_name, uppercase_first_letter=True)
|
|
contents = migration_template.format(migration_name)
|
|
time = datetime.utcnow().strftime("%Y%m%d%H%M%S")
|
|
migration_file_name = "{}_{}.rb".format(time, migration_base_name)
|
|
with open(os.path.join("src/db/addon/", skeleton["name"], migration_file_name), 'w') as f:
|
|
f.write(contents)
|
|
|
|
|
|
main()
|