New files added

Author: iGSRi

docs/database_loading.png

@@ -0,0 +1 @@
+from .ln2sql import Ln2sql

docs/fr2sql_recital_2015_paper.pdf

@@ -0,0 +1,50 @@
+class Column:
+ def __init__(self, name='', type=None, equivalences=None):
+ self._name = name
+
+ if not type:
+ type = []
+ self._type = type
+
+ if not equivalences:
+ equivalences = []
+ self._equivalences = equivalences
+
+ self.primary = False
+ self.foreign = False
+
+ @property
+ def name(self):
+ return self._name
+
+ @property
+ def type(self):
+ return self._type
+
+ def add_type(self, type):
+ self.type.append(type)
+
+ @property
+ def equivalences(self):
+ return self._equivalences
+
+ def add_equivalence(self, equivalence):
+ self.equivalences.append(equivalence)
+
+ def is_equivalent(self, word):
+ if word in self.equivalences:
+ return True
+ else:
+ return False
+
+ def is_primary(self):
+ return self.primary
+
+ def set_as_primary(self):
+ self.primary = True
+
+ def is_foreign(self):
+ return self.foreign
+
+ def set_as_foreign(self, references):
+ self.foreign = references

docs/graphical_user_interface.png

@@ -0,0 +1,23 @@
+class Color:
+ PURPLE = '\033[95m'
+ CYAN = '\033[96m'
+ DARKCYAN = '\033[36m'
+ BLUE = '\033[94m'
+ GREEN = '\033[92m'
+ YELLOW = '\033[93m'
+ RED = '\033[91m'
+ BOLD = '\033[1m'
+ UNDERLINE = '\033[4m'
+ END = '\033[0m'
+ 
+def without_color():
+ Color.PURPLE = ''
+ Color.CYAN = ''
+ Color.DARKCYAN = ''
+ Color.BLUE = ''
+ Color.GREEN = ''
+ Color.YELLOW = ''
+ Color.RED = ''
+ Color.BOLD = ''
+ Color.UNDERLINE = ''
+ Color.END = ''

docs/mvc_class_diagram.png

@@ -0,0 +1,158 @@
+import os
+import re
+
+from .constants import Color
+from .table import Table
+
+
+class Database:
+
+ def __init__(self):
+ self.tables = []
+ self.thesaurus_object = None
+
+ def set_thesaurus(self, thesaurus):
+ self.thesaurus_object = thesaurus
+
+ def get_number_of_tables(self):
+ return len(self.tables)
+
+ def get_tables(self):
+ return self.tables
+
+ def get_column_with_this_name(self, name):
+ for table in self.tables:
+ for column in table.get_columns():
+ if column.name == name:
+ return column
+
+ def get_table_by_name(self, table_name):
+ for table in self.tables:
+ if table.name == table_name:
+ return table
+
+ def get_tables_into_dictionary(self):
+ data = {}
+ for table in self.tables:
+ data[table.name] = []
+ for column in table.get_columns():
+ data[table.name].append(column.name)
+ return data
+
+ def get_primary_keys_by_table(self):
+ data = {}
+ for table in self.tables:
+ data[table.name] = table.get_primary_keys()
+ return data
+
+ def get_foreign_keys_by_table(self):
+ data = {}
+ for table in self.tables:
+ data[table.name] = table.get_foreign_keys()
+ return data
+
+ def get_primary_keys_of_table(self, table_name):
+ for table in self.tables:
+ if table.name == table_name:
+ return table.get_primary_keys()
+
+ def get_primary_key_names_of_table(self, table_name):
+ for table in self.tables:
+ if table.name == table_name:
+ return table.get_primary_key_names()
+
+ def get_foreign_keys_of_table(self, table_name):
+ for table in self.tables:
+ if table.name == table_name:
+ return table.get_foreign_keys()
+
+ def get_foreign_key_names_of_table(self, table_name):
+ for table in self.tables:
+ if table.name == table_name:
+ return table.get_foreign_key_names()
+
+ def add_table(self, table):
+ self.tables.append(table)
+
+ @staticmethod
+ def _generate_path(path):
+ cwd = os.path.dirname(__file__)
+ filename = os.path.join(cwd, path)
+ return filename
+
+ def load(self, path):
+ with open(self._generate_path(path)) as f:
+ content = f.read()
+ tables_string = [p.split(';')[0] for p in content.split('CREATE') if ';' in p]
+ for table_string in tables_string:
+ if 'TABLE' in table_string:
+ table = self.create_table(table_string)
+ self.add_table(table)
+ alter_tables_string = [p.split(';')[0] for p in content.split('ALTER') if ';' in p]
+ for alter_table_string in alter_tables_string:
+ if 'TABLE' in alter_table_string:
+ self.alter_table(alter_table_string)
+
+ def predict_type(self, string):
+ if 'int' in string.lower():
+ return 'int'
+ elif 'char' in string.lower() or 'text' in string.lower():
+ return 'string'
+ elif 'date' in string.lower():
+ return 'date'
+ else:
+ return 'unknow'
+
+ def create_table(self, table_string):
+ lines = table_string.split("\n")
+ table = Table()
+ for line in lines:
+ if 'TABLE' in line:
+ table_name = re.search("`(\w+)`", line)
+ table.name = table_name.group(1)
+ if self.thesaurus_object is not None:
+ table.equivalences = self.thesaurus_object.get_synonyms_of_a_word(table.name)
+ elif 'PRIMARY KEY' in line:
+ primary_key_columns = re.findall("`(\w+)`", line)
+ for primary_key_column in primary_key_columns:
+ table.add_primary_key(primary_key_column)
+ else:
+ column_name = re.search("`(\w+)`", line)
+ if column_name is not None:
+ column_type = self.predict_type(line)
+ if self.thesaurus_object is not None:
+ equivalences = self.thesaurus_object.get_synonyms_of_a_word(column_name.group(1))
+ else:
+ equivalences = []
+ table.add_column(column_name.group(1), column_type, equivalences)
+ return table
+
+ def alter_table(self, alter_string):
+ lines = alter_string.replace('\n', ' ').split(';')
+ for line in lines:
+ if 'PRIMARY KEY' in line:
+ table_name = re.search("TABLE `(\w+)`", line).group(1)
+ table = self.get_table_by_name(table_name)
+ primary_key_columns = re.findall("PRIMARY KEY \(`(\w+)`\)", line)
+ for primary_key_column in primary_key_columns:
+ table.add_primary_key(primary_key_column)
+ elif 'FOREIGN KEY' in line:
+ table_name = re.search("TABLE `(\w+)`", line).group(1)
+ table = self.get_table_by_name(table_name)
+ foreign_keys_list = re.findall("FOREIGN KEY \(`(\w+)`\) REFERENCES `(\w+)` \(`(\w+)`\)", line)
+ for column, foreign_table, foreign_column in foreign_keys_list:
+ table.add_foreign_key(column, foreign_table, foreign_column)
+
+ def print_me(self):
+ for table in self.tables:
+ print('+-------------------------------------+')
+ print("| %25s |" % (table.name.upper()))
+ print('+-------------------------------------+')
+ for column in table.columns:
+ if column.is_primary():
+ print("| 🔑 %31s |" % (Color.BOLD + column.name + ' (' + column.get_type() + ')' + Color.END))
+ elif column.is_foreign():
+ print("| #️⃣ %31s |" % (Color.ITALIC + column.name + ' (' + column.get_type() + ')' + Color.END))
+ else:
+ print("| %23s |" % (column.name + ' (' + column.get_type() + ')'))
+ print('+-------------------------------------+\n')