commit 21e306191984478b43c667cd26337bfa89071065 Author: Хирецкий Михаил Date: Mon Oct 18 11:50:10 2010 +0400 Initial commit diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..d645695 --- /dev/null +++ b/LICENSE @@ -0,0 +1,202 @@ + + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + + TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + + 1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + + 2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + + 3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + + 4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + + 5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + + 6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + + 7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + + 8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + + 9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + + END OF TERMS AND CONDITIONS + + APPENDIX: How to apply the Apache License to your work. + + To apply the Apache License to your work, attach the following + boilerplate notice, with the fields enclosed by brackets "[]" + replaced with your own identifying information. (Don't include + the brackets!) The text should be enclosed in the appropriate + comment syntax for the file format. We also recommend that a + file or class name and description of purpose be included on the + same "printed page" as the copyright notice for easier + identification within third-party archives. + + Copyright [yyyy] [name of copyright owner] + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. diff --git a/README b/README new file mode 100644 index 0000000..2e4d1ec --- /dev/null +++ b/README @@ -0,0 +1,13 @@ +AUTHOR: Calculate Ltd. + +INSTALL +------- + +calculate-assemble needs the following library version installed, in order to run: + calculate-install >= 2.2.0.1 + calculate-lib >= 2.2.0.1 + +To install calculate-assemble , just execute the install script 'setup.py'. +Example: + + ./setup.py install diff --git a/i18n/cl_assemble_ru.mo b/i18n/cl_assemble_ru.mo new file mode 100644 index 0000000..b708136 Binary files /dev/null and b/i18n/cl_assemble_ru.mo differ diff --git a/pym/__init__.py b/pym/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/pym/cl_assemble.py b/pym/cl_assemble.py new file mode 100644 index 0000000..df8f6b0 --- /dev/null +++ b/pym/cl_assemble.py @@ -0,0 +1,184 @@ +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +__version__ = "2.2.0.1" +__app__ = "calculate-assemble" + +import os +import re +import sys +import traceback +from os import path +from cl_template import template +from cl_utils import process,pathJoin,getRunCommands,getTupleVersion +from subprocess import STDOUT,PIPE +from cl_print import color_print +from cl_datavars import DataVars +from cl_distr import IsoDistributive, DirectoryDistributive, \ + DistributiveError +from cl_vars_share import varsShare + +from cl_lang import lang +from server.utils import dialogYesNo +import cl_overriding + +class printNoColor: + def colorPrint(self,attr,fg,bg,string): + sys.stdout.write(string) + +class AssembleError(Exception): + """Installation Error""" + assemble_errors = "" + + def installExit(self,*args,**kwars): + raise AssembleError(self.__class__.assemble_errors) + + def overprintERROR(self,error): + self.__class__.assemble_errors += str(error) + "\n" + + def getAssembleErrors(self): + return self.__class__.assemble_errors + + def popAssembleErrors(self): + res = self.__class__.assemble_errors + self.__class__.assemble_errors = "" + return res + + def getOverrideMethods(self): + return self.installExit, self.overprintERROR + +cl_overriding.exit, cl_overriding.printERROR = \ + AssembleError().getOverrideMethods() + +class DataVarsAssemble(DataVars): + """Variable class for assemble""" + + def importAssemble(self, **args): + '''Get variables for assemble''' + # section name in calculate.env + envSection = "assemble" + # import assemble variables + self.importData(envSection, ('cl_vars_assemble','cl_fill_assemble')) + +class cl_assemble(color_print): + """Primary class for assemble actions""" + + def __init__(self): + self.clVars = None + self.startMessage = "" + self.lenStartMessage = 0 + self.clTempl = None + self.force = False + + def setNoColor(self): + self.color = False + + def initVars(self): + """Primary initialization of variables""" + self.clVars = DataVarsAssemble() + self.clVars.importAssemble() + self.clVars.flIniFile() + + def applyTemplatesForSquash(self,directory): + """Apply templates for root of system.""" + #self.clVars.Set("cl_root_path","/", True) + self.clVars.Set("cl_action","squash", True) + self.clVars.Set("cl_chroot_path",directory, True) + self.clTempl = template(self.clVars) + dirsFiles = self.clTempl.applyTemplates() + self.clTempl.closeFiles() + if self.clTempl.getError(): + raise AssembleError(self.clTempl.getError()) + else: + return dirsFiles + + def printMessageForTest(self, message, lenMsg=False): + """Print waiting message and OK or Error by func result""" + message = "%s ..." % message + self.printSUCCESS(message,printBR=False) + self.startMessage = message + if lenMsg: + self.lenStartMessage = lenMsg + else: + self.lenStartMessage = self.lenString(self.startMessage) + + def printOnlyNotOK(self, string, offsetL=0, printBR=True): + """Вывод на печать в случае сбоя""" + self._printSysOut = sys.stdout + self.printLine((('', string),), + (('blueBr','['), + ('redBr',' !! '), + ('blueBr',']'), + ), offsetL, printBR) + + def printByResult(self,result,failMessage=None): + if self.startMessage: + offset = 3 + if result: + self.printOnlyOK(" ", self.lenStartMessage + offset) + else: + self.printOnlyNotOK(" ", self.lenStartMessage + offset) + if failMessage: + self.printERROR(failMessage) + self.startMessage = "" + self.lenStartMessage = 0 + + def make(self,logicFunc,*argv): + """Make iso image by variables""" + self.sourceDistr = None + self.targetDistr = None + error = None + try: + if not logicFunc(*argv): + return False + except (EOFError), e: + error = e + except (AssembleError,DistributiveError),e: + error = e + except (Exception),e: + error = "" + for i in apply(traceback.format_exception, sys.exc_info()): + error += i + except KeyboardInterrupt,e: + self.printByResult(False) + self.printWARNING("Interrupting the system assembling") + error = _("System assembling manually interrupt") + if error: + self.printByResult(False) + try: + if self.clTempl: + self.clTempl.closeFiles() + if self.sourceDistr: + self.printMessageForTest(_("Releasing source data")) + self.sourceDistr.close() + self.printByResult(True) + if self.targetDistr: + self.printMessageForTest(_("Unmount assembling volume")) + self.targetDistr.close() + self.printByResult(True) + except (AssembleError,DistributiveError),e: + error = "%s\n%s" % (str(error),_("Unmounting error")) + except KeyboardInterrupt,e: + pass + if error: + self.printByResult(False) + if error: + for line in filter(lambda x: x,str(error).split('\n')): + self.printERROR(line) + self.printERROR(_("System assembling failed")) + return False + self.printSUCCESS(_("System was have assembled successfully")) + return True diff --git a/pym/cl_assemble_cmd.py b/pym/cl_assemble_cmd.py new file mode 100644 index 0000000..433aa6b --- /dev/null +++ b/pym/cl_assemble_cmd.py @@ -0,0 +1,55 @@ +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from cl_assemble import cl_assemble, __app__, __version__ +from cl_opt import opt +from cl_share_cmd import share_cmd +import os +from os import path, access, R_OK +import re +import sys + +from cl_lang import lang +lang().setLanguage(sys.modules[__name__]) + +DESCRIPTION = _("The Calculate Linux system assemble") +CMD_OPTIONS = [{'shortOption':"f", + 'longOption':"force", + 'help':_("no questions during the creating process") + }] +USAGE = _("%prog [options]") + +class assemble_cmd(share_cmd): + """Class for work with cl_assemble by console""" + def __init__(self): + self.optobj = opt(package=__app__, + version=__version__, + description=DESCRIPTION, + usage=USAGE, + option_list= CMD_OPTIONS + opt.variable_control + + opt.color_control, + check_values=self.checkOpts) + self.logicObj = cl_assemble() + #self.optionsInitrdIncompatible = ["o","no_clean","m","mdadm","lvm", + # "k", "e","dmraid","c","ebuild", + # "symlink"] + + def checkOpts(self, values, args): + """Check values all specified options.""" + if len(args) > 0: + self.optobj.error(_("unrecognized option") + ": %s"% "".join(args)) + self.optobj.checkVarSyntax(values) + return (values, args) diff --git a/pym/cl_fill_assemble.py b/pym/cl_fill_assemble.py new file mode 100644 index 0000000..538b226 --- /dev/null +++ b/pym/cl_fill_assemble.py @@ -0,0 +1,24 @@ +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import os +import re +import cl_overriding +from cl_vars_share import varsShare +from os import path + +class fillVars(object, varsShare): + pass \ No newline at end of file diff --git a/pym/cl_share_cmd.py b/pym/cl_share_cmd.py new file mode 100644 index 0000000..d762259 --- /dev/null +++ b/pym/cl_share_cmd.py @@ -0,0 +1,93 @@ +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import sys +import os + +from cl_print import color_print +from cl_utils import _error + +# Application message translator +from cl_lang import lang +lang().setLanguage(sys.modules[__name__]) +import cl_overriding + +class share_cmd(color_print, _error): + """Class of share methods for perform cmd line""" + def isRoot(self, printError=True): + """Detect root user""" + if os.getuid() == 0 and os.getgid() == 0: + return True + else: + if printError: + self.printERROR(_("The user is not root")) + return False + + def printVars(self, optObj): + """Variable printing from cmdline""" + if optObj.v: + varsFilter = None + varsNames = [] + format = "default" + # variable filter + if optObj.filter: + optCmd = optObj.filter + if ',' in optCmd: + varsNames = optCmd.split(",") + elif '*' in optCmd: + varsFilter = optCmd.replace("*", ".*") + else: + varsNames.append(optCmd) + if optObj.xml: + format = "xml" + try: + v = int(optObj.v) + except: + v = 1 + self.logicObj.clVars.printVars(varsFilter, varsNames, + outFormat=format, + verbose=v) + + def setVars(self, optObj): + """Variable setting from cmdline""" + if optObj.set: + for vals in optObj.set: + for val in vals.split(','): + k,o,v = val.partition('=') + if self.logicObj.clVars.exists(k): + if self.logicObj.clVars.SetWriteVar(k,v) == False: + return False + else: + self.printERROR(_('variable %s not found')%k) + return False + return True + + def writeVars(self, optObj): + """Variable writing""" + if not self.logicObj.clVars.WriteVars(header="install"): + errMsg = self.getError() + if errMsg: + self.printERROR(errMsg.strip()) + self.printERROR(_('Can not write template variables')) + return False + return True + + def setPrintNoColor(self, optObj): + """Setting message printing without color""" + if optObj.color and optObj.color=="never": + color_print.colorPrint = lambda *arg : sys.stdout.write(arg[-1]) or\ + sys.stdout.flush() + diff --git a/pym/cl_vars_assemble.py b/pym/cl_vars_assemble.py new file mode 100644 index 0000000..5af97f1 --- /dev/null +++ b/pym/cl_vars_assemble.py @@ -0,0 +1,32 @@ +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 # +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +#Available values of keys +# mode - read only or writeable variable +# value - default variable value +# select - list of posible values for variable +# hide - flag, if it is True, then the variable is not printable +# printval - print value of variable +from cl_assemble import __version__, __app__ + +class Data: + # relative path for apply templates on files of system + cl_root_path = {} + + # program name + cl_name = {'value':__app__} + + # program version + cl_ver = {'value':__version__} diff --git a/scripts/cl-assemble b/scripts/cl-assemble new file mode 100644 index 0000000..de5f808 --- /dev/null +++ b/scripts/cl-assemble @@ -0,0 +1,51 @@ +#!/usr/bin/python +#-*- coding: utf-8 -*- + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import sys +import os +from os import path +sys.path.insert(0, path.abspath('/usr/lib/calculate-2.2/calculate-lib/pym')) +sys.path.insert(0, path.abspath('/usr/lib/calculate-2.2/calculate-install/pym')) +sys.path.insert(0, path.abspath('/usr/lib/calculate-2.2/calculate-assemble/pym')) + +from cl_lang import lang +tr = lang() +tr.setGlobalDomain('cl_assemble') +tr.setLanguage(sys.modules[__name__]) + +from cl_assemble_cmd import assemble_cmd + +if __name__ == "__main__": + assemble = assemble_cmd() + assemble.logicObj.initVars() + ret = assemble.optobj.parse_args() + if ret is False: + sys.exit(1) + options, args = ret + # set color/nocolor for display messages + assemble.setPrintNoColor(options) + # set values to variables + if not assemble.setVars(options): + sys.exit(1) + # print variables + if options.v or options.filter or options.xml: + assemble.printVars(options) + sys.exit(0) + # check root + if not assemble.isRoot(): + sys.exit(1) + sys.exit(0) diff --git a/setup.cfg b/setup.cfg new file mode 100644 index 0000000..709f35e --- /dev/null +++ b/setup.cfg @@ -0,0 +1,5 @@ +[install] +install-scripts=/usr/bin +install-purelib=/usr/lib/calculate-2.2 +install-platlib=/usr/lib/calculate-2.2 +#install-data=/usr/lib/calculate-2.2/calculate-assemble diff --git a/setup.py b/setup.py new file mode 100755 index 0000000..a8cc375 --- /dev/null +++ b/setup.py @@ -0,0 +1,112 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +# setup.py --- Setup script for calculate-assemble + +# Copyright 2010 Calculate Ltd. http://www.calculate-linux.org +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import os +import stat +from distutils.core import setup, Extension +from distutils.command.install_data import install_data + +data_files = [] + +var_data_files = [] + +#data_dirs_template = ['templates'] +share_calculate_dir = "/usr/share/calculate" +data_dirs_share = ['i18n'] +#template_calculate_dir = os.path.join(share_calculate_dir, "templates") +#template_replace_dirname = "install" + + +def __scanDir(scanDir, prefix, replace_dirname, dirData, flagDir=False): + """Scan directory""" + files = [] + dirs = [] + if flagDir or stat.S_ISDIR(os.stat(scanDir)[stat.ST_MODE]): + for fileOrDir in os.listdir(scanDir): + absPath = os.path.join(scanDir,fileOrDir) + statInfo = os.stat(absPath)[stat.ST_MODE] + if stat.S_ISREG(statInfo): + files.append(absPath) + elif stat.S_ISDIR(statInfo): + dirs.append(absPath) + if replace_dirname: + listDirs = list(scanDir.partition("/"))[1:] + listDirs.insert(0,replace_dirname) + scanDir = "".join(listDirs) + if prefix: + scanDir = os.path.join(prefix,scanDir) + dirData.append((scanDir, files)) + for sDir in dirs: + __scanDir(sDir, prefix, replace_dirname,dirData, True) + return dirData + +def create_data_files(data_dirs, prefix="", replace_dirname=""): + """Create data_files""" + data_files = [] + for data_dir in data_dirs: + data = [] + data_files += __scanDir(data_dir, prefix, replace_dirname, data) + return data_files + +#data_files += create_data_files (data_dirs_template,template_calculate_dir, +# template_replace_dirname) +data_files += create_data_files (data_dirs_share, share_calculate_dir) + +class cl_install_data(install_data): + def run (self): + install_data.run(self) + #data_file = [("/etc/init.d/calculate-2.2",0755)] + data_file = [] + fileNames = map(lambda x: os.path.split(x[0])[1], data_file) + listNames = map(lambda x: filter(lambda y: y, x[0].split("/")), + data_file) + data_find = {} + for i in range(len(fileNames)): + listNames[i].reverse() + data_find[fileNames[i]] =[listNames[i],data_file[i][1]] + + for path in self.get_outputs(): + nameFile = os.path.split(path)[1] + if nameFile in data_find.keys(): + data = data_find[nameFile][0] + mode = data_find[nameFile][1] + flagFound = True + iMax = len(data) + pathFile = path + for i in range(iMax): + if data[i] != os.path.split(pathFile)[1]: + flagFound = False + break + pathFile = os.path.split(pathFile)[0] + if flagFound: + os.chmod(path, mode) + +setup( + name = 'calculate-assemble', + version = "2.2.0.1", + description = "System assembling for Calculate Linux", + author = "Calculate Ltd.", + author_email = "support@calculate.ru", + url = "http://calculate-linux.org", + license = "http://www.apache.org/licenses/LICENSE-2.0", + package_dir = {'calculate-assemble': "."}, + packages = ['calculate-assemble.pym'], + data_files = data_files, + scripts=["./scripts/cl-assemble"], +)