mirror of
https://github.com/rbock/sqlpp11.git
synced 2024-11-15 20:31:16 +08:00
Merge pull request #233 from egorpugin/develop
Add simplified sqlite2cpp generator script that uses sqlite3 schema parsing.
This commit is contained in:
Roland Bock
GitHub
commit
f204bf0b9d
199
scripts/sqlite2cpp.py
Normal file
199
scripts/sqlite2cpp.py
Normal file
@ -0,0 +1,199 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
##
|
||||
# Copyright (c) 2013-2015, Roland Bock
|
||||
# Copyright (c) 2018, Egor Pugin
|
||||
# All rights reserved.
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without modification,
|
||||
# are permitted provided that the following conditions are met:
|
||||
#
|
||||
# * Redistributions of source code must retain the above copyright notice,
|
||||
# this list of conditions and the following disclaimer.
|
||||
# * Redistributions in binary form must reproduce the above copyright notice,
|
||||
# this list of conditions and the following disclaimer in the documentation
|
||||
# and/or other materials provided with the distribution.
|
||||
#
|
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
||||
# IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
|
||||
# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
||||
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
|
||||
# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
|
||||
# OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
##
|
||||
|
||||
import argparse
|
||||
import os
|
||||
import re
|
||||
import sqlite3
|
||||
import sys
|
||||
|
||||
INCLUDE = 'sqlpp11'
|
||||
NAMESPACE = 'sqlpp'
|
||||
|
||||
# map sqlite3 types
|
||||
types = {
|
||||
'integer': 'integer',
|
||||
'text': 'text',
|
||||
'blob': 'blob',
|
||||
'real': 'floating_point',
|
||||
}
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='sqlpp11 cpp schema generator')
|
||||
parser.add_argument('ddl', help='path to ddl')
|
||||
parser.add_argument('target', help='path to target')
|
||||
parser.add_argument('namespace', help='namespace')
|
||||
args = parser.parse_args()
|
||||
|
||||
pathToHeader = args.target + '.h'
|
||||
|
||||
# execute schema scripts
|
||||
conn = sqlite3.connect(':memory:')
|
||||
conn.executescript(open(args.ddl).read())
|
||||
|
||||
# set vars
|
||||
toClassName = class_name_naming_func
|
||||
toMemberName = member_name_naming_func
|
||||
DataTypeError = False
|
||||
|
||||
header = open(pathToHeader, 'w')
|
||||
namespace = args.namespace
|
||||
nsList = namespace.split('::')
|
||||
|
||||
# start printing
|
||||
print('// generated by ' + ' '.join(sys.argv), file=header)
|
||||
print('#ifndef '+get_include_guard_name(namespace, pathToHeader), file=header)
|
||||
print('#define '+get_include_guard_name(namespace, pathToHeader), file=header)
|
||||
print('', file=header)
|
||||
print('#include <' + INCLUDE + '/table.h>', file=header)
|
||||
print('#include <' + INCLUDE + '/data_types.h>', file=header)
|
||||
print('#include <' + INCLUDE + '/char_sequence.h>', file=header)
|
||||
print('', file=header)
|
||||
for ns in nsList:
|
||||
print('namespace ' + ns, file=header)
|
||||
print('{', file=header)
|
||||
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
|
||||
tables = cursor.fetchall()
|
||||
for table_name in tables:
|
||||
table_name = table_name[0]
|
||||
|
||||
if table_name.startswith('sqlite_'):
|
||||
continue
|
||||
|
||||
sqlTableName = table_name
|
||||
tableClass = toClassName(sqlTableName)
|
||||
tableMember = toMemberName(sqlTableName)
|
||||
tableNamespace = tableClass + '_'
|
||||
tableTemplateParameters = tableClass
|
||||
print(' namespace ' + tableNamespace, file=header)
|
||||
print(' {', file=header)
|
||||
|
||||
columns = cursor.execute("PRAGMA table_info('%s')" % table_name).fetchall()
|
||||
for column in columns:
|
||||
sqlColumnName = column[1]
|
||||
columnClass = toClassName(sqlColumnName)
|
||||
tableTemplateParameters += ',\n ' + tableNamespace + '::' + columnClass
|
||||
columnMember = toMemberName(sqlColumnName)
|
||||
sqlColumnType = column[2].lower()
|
||||
print(' struct ' + columnClass, file=header)
|
||||
print(' {', file=header)
|
||||
print(' struct _alias_t', file=header)
|
||||
print(' {', file=header)
|
||||
print(' static constexpr const char _literal[] = "' + escape_if_reserved(sqlColumnName) + '";', file=header)
|
||||
print(' using _name_t = sqlpp::make_char_sequence<sizeof(_literal), _literal>;', file=header)
|
||||
print(' template<typename T>', file=header)
|
||||
print(' struct _member_t', file=header)
|
||||
print(' {', file=header)
|
||||
print(' T ' + columnMember + ';', file=header)
|
||||
print(' T& operator()() { return ' + columnMember + '; }', file=header)
|
||||
print(' const T& operator()() const { return ' + columnMember + '; }', file=header)
|
||||
print(' };', file=header)
|
||||
print(' };', file=header)
|
||||
try:
|
||||
traitslist = [NAMESPACE + '::' + types[sqlColumnType]]
|
||||
except KeyError as e:
|
||||
print ('Error: datatype "' + sqlColumnType + '"" is not supported.')
|
||||
DataTypeError = True
|
||||
requireInsert = True
|
||||
hasAutoValue = sqlColumnName == 'id'
|
||||
if hasAutoValue:
|
||||
traitslist.append(NAMESPACE + '::tag::must_not_insert')
|
||||
traitslist.append(NAMESPACE + '::tag::must_not_update')
|
||||
requireInsert = False
|
||||
if column[3] == '0':
|
||||
traitslist.append(NAMESPACE + '::tag::can_be_null')
|
||||
requireInsert = False
|
||||
if column[4]:
|
||||
requireInsert = False
|
||||
if requireInsert:
|
||||
traitslist.append(NAMESPACE + '::tag::require_insert')
|
||||
print(' using _traits = ' + NAMESPACE + '::make_traits<' + ', '.join(traitslist) + '>;', file=header)
|
||||
print(' };', file=header)
|
||||
|
||||
print(' } // namespace ' + tableNamespace, file=header)
|
||||
print('', file=header)
|
||||
|
||||
print(' struct ' + tableClass + ': ' + NAMESPACE + '::table_t<' + tableTemplateParameters + '>', file=header)
|
||||
print(' {', file=header)
|
||||
print(' struct _alias_t', file=header)
|
||||
print(' {', file=header)
|
||||
print(' static constexpr const char _literal[] = "' + sqlTableName + '";', file=header)
|
||||
print(' using _name_t = sqlpp::make_char_sequence<sizeof(_literal), _literal>;', file=header)
|
||||
print(' template<typename T>', file=header)
|
||||
print(' struct _member_t', file=header)
|
||||
print(' {', file=header)
|
||||
print(' T ' + tableMember + ';', file=header)
|
||||
print(' T& operator()() { return ' + tableMember + '; }', file=header)
|
||||
print(' const T& operator()() const { return ' + tableMember + '; }', file=header)
|
||||
print(' };', file=header)
|
||||
print(' };', file=header)
|
||||
print(' };', file=header)
|
||||
|
||||
|
||||
for ns in nsList:
|
||||
print('} // namespace ' + ns, file=header)
|
||||
print('#endif', file=header)
|
||||
|
||||
if (DataTypeError):
|
||||
print("Error: unsupported datatypes." )
|
||||
print("Possible solutions:")
|
||||
print("A) Implement this datatype (examples: sqlpp11/data_types)" )
|
||||
print("B) Extend/upgrade ddl2cpp (edit types map)" )
|
||||
print("C) Raise an issue on github" )
|
||||
sys.exit(10) #return non-zero error code, we might need it for automation
|
||||
|
||||
def get_include_guard_name(namespace, inputfile):
|
||||
val = re.sub("[^A-Za-z0-9]+", "_", namespace + '_' + os.path.basename(inputfile))
|
||||
return val.upper()
|
||||
|
||||
def repl_camel_case_func(m):
|
||||
if m.group(1) == '_':
|
||||
return m.group(2).upper()
|
||||
else:
|
||||
return m.group(1) + m.group(2).upper()
|
||||
|
||||
def class_name_naming_func(s):
|
||||
return re.sub("(^|\s|[_0-9])(\S)", repl_camel_case_func, s)
|
||||
|
||||
def member_name_naming_func(s):
|
||||
return re.sub("(\s|_|[0-9])(\S)", repl_camel_case_func, s)
|
||||
|
||||
def escape_if_reserved(name):
|
||||
reserved_names = [
|
||||
'GROUP',
|
||||
'ORDER',
|
||||
]
|
||||
if name.upper() in reserved_names:
|
||||
return '!{}'.format(name)
|
||||
return name
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
sys.exit(0)
|
||||
Loading…
Reference in New Issue
Block a user