X-Git-Url: https://code.agnibho.com/repo?p=ddstorm.git;a=blobdiff_plain;f=compile.py;h=80bb4d1456f07621b9e71890592f309c14b69768;hp=a94e73048879d4cc7da284e979f4fb270be77597;hb=HEAD;hpb=49346f79fdffaccfa0a65dc0ba412311aa27a668
diff --git a/compile.py b/compile.py
index a94e730..80bb4d1 100644
--- a/compile.py
+++ b/compile.py
@@ -1,71 +1,113 @@
#! /usr/bin/python3
-# DDStorm
-# -------
-# Copyright (c) 2015 Agnibho Mondal
-# All rights reserved
+'''
+This module converts text differetial diagnosis files
+to a simplified modular file format that can be used
+by the program.
+'''
+'''
+Copyright (c) 2015 Agnibho Mondal
+All rights reserved
-# This file is part of DDStorm.
+This file is part of DDStorm.
-# DDStorm is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
+DDStorm is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
-# DDStorm is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU General Public License for more details.
+DDStorm is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+GNU General Public License for more details.
-# You should have received a copy of the GNU General Public License
-# along with DDStorm. If not, see .
+You should have received a copy of the GNU General Public License
+along with DDStorm. If not, see .
+'''
-import os, logging, re
+import os
+import logging
+import re
from operator import itemgetter
from fnmatch import fnmatch
+
from conf import Conf
from const import *
from alias import Alias
+
logging.basicConfig(filename=LOG_FILE)
class Compile:
+ '''
+ This class creates a compiler for the DDStorm
+ that compiles the text files containing list of
+ differential diagnosis to simplified modular
+ data files usable by the program.
+ '''
+
def __init__(self, conf=False):
- self.source=set()
- self.custom=set()
+ '''
+ The constructor optionally accepts a configuration.
+ If none is provided it creates a default configuration.
+
+ Parameters:
+ conf - A dictionary containing configuration options
+ '''
if(conf):
- self.conf=conf
+ self._conf=conf
else:
- self.conf=Conf()
- self.alias=Alias(conf)
+ self._conf=Conf()
self.clean=True
- for path, subdirs, files in os.walk(self.conf.get("library_path")):
+
+ def compile(self):
+ ''' Compile the text files to DDStorm modules. '''
+ self.source=set()
+ self.custom=set()
+ self.alias=Alias(self._conf)
+
+ # Loop over library files and add *.txt files to source
+ for path, subdirs, files in os.walk(self._conf.get("library_path")):
for name in files:
if(fnmatch(name, "*.txt")):
self.source.add(os.path.join(path, name))
- for path, subdirs, files in os.walk(self.conf.get("custom_path")):
+
+ # Loop over custom files and add *.txt files to custom
+ for path, subdirs, files in os.walk(self._conf.get("custom_path")):
for name in files:
if(fnmatch(name, "*.txt")):
self.custom.add(os.path.join(path, name))
- if(not os.path.isdir(self.conf.get("module_path"))):
- os.makedirs(self.conf.get("module_path"))
- for f in os.listdir(self.conf.get("module_path")):
+
+ # Create module directory if not already present and delete all module files
+ if(not os.path.isdir(self._conf.get("module_path"))):
+ os.makedirs(self._conf.get("module_path"))
+ for f in os.listdir(self._conf.get("module_path")):
if(fnmatch(f, "*.module")):
- os.unlink(self.conf.get("module_path")+f)
+ os.unlink(self._conf.get("module_path")+f)
+
+ # Create a regex for calculating priority from filename
self.priorityRegex=re.compile("(?<=\.)\d+$")
+
+ # First sort files by priority then compile them to module
for src in self._sortPriority(self.source):
self._makeModule(src)
for src in self._sortPriority(self.custom):
self._makeModule(src)
def _sortPriority(self, files):
+ ''' Sort data files based on their priority settings. '''
ls=[]
+ # Loop over the files
for addr in files:
+ # Format the file name
name=os.path.splitext(os.path.basename(addr))[0].lower().replace("_"," ").replace("-", " ")
+ # Search for priority tag on file name
m=re.search(self.priorityRegex, name)
+ # Add to ls as (symptom name, priority number, file name) with default priority of 100
if(m):
ls.append((name.replace("."+m.group(), ""), int(m.group()), addr))
else:
ls.append((name, 100, addr))
+ # Sort the file list, first by the symptom name, then by the priority number
ls.sort(reverse=True)
if(ls):
return(list(zip(*ls))[2])
@@ -73,13 +115,19 @@ class Compile:
return ls
def _makeModule(self, src):
+ ''' Create application usable modules from data files. '''
+ # Format the file name
module=os.path.splitext(os.path.basename(src))[0].lower().replace("_"," ").replace("-", " ")
+ # Remove the priority tag from file name
m=re.search(self.priorityRegex, module)
if(m):
module=module.replace("."+m.group(), "")
- modFile=self.conf.get("module_path")+module+".module"
+ # Create the module file name
+ modFile=self._conf.get("module_path")+module+".module"
modFlag=False
+ # Loop over both files, the source data file and the target module file
with open(src, "r") as sf, open(modFile, "a") as tf:
+ # Ignore lines starting with ! or #, + and - has special meaning, write other lines to module. Log the errors.
for line in sf:
line=line.strip().split("#")[0]
if(len(line)==0):
@@ -97,17 +145,20 @@ class Compile:
else:
self.clean=False
logging.warning("Syntax error in file '"+src+"': "+line)
+ # Deal with special lines
if(modFlag):
modFlag=False
with open(src, "r") as f:
for line in f:
line=line.strip().split("#")[0]
if(line[1:].replace(" ","").replace("-","").replace("_","").replace("'","").isalnum()):
+ # If line starts with + add it to the module file
if(line.startswith("+")):
with open(modFile, "r") as fn:
text=fn.read()
with open(modFile, "w") as fn:
print(self.alias.get(line[1:]).capitalize()+"\n"+text, file=fn)
+ # If line starts with - remove corresponding item from the module file
elif(line.startswith("-")):
with open(modFile, "r") as fn:
text=fn.read()
@@ -115,8 +166,13 @@ class Compile:
with open(modFile, "w") as fn:
print(text, file=fn)
+ def is_clean(self):
+ '''Report if compilation ended successfully'''
+ return self.clean
+
def main():
- c=Compile()
+ ''' Compile the data files into formatted module files '''
+ c=Compile().compile()
if(__name__=="__main__"):
main()