Exemplo n.º 1
0
def flatten( dir, mfd ):
  # pull chunks up to chapter level
  # dir is like: /tmp/input/<user>/<repo>/
  # content is under dir [<chapter>]<chunk>.[md|txt]
    myLog( 'info', "flatten: " + dir + " mfd: " + mfd )
    os.chdir( dir )
    content =  os.path.join( dir, "content" )

    if os.path.exists( content ):
        os.chdir( content )

    mdFiles = glob( '*/*.md' ) 
    make_dir( mfd )
    print( "mdFiles" )
    print( mdFiles )
    fileCount = 0

    for mdFile in mdFiles:
        newFile = mdFile.replace( '/', '-' )
        os.rename( mdFile, os.path.join( mfd,newFile ))
        myLog( "debug", "mdFile: " + mdFile + " newFile: " + newFile )
        fileCount += 1

    myLog( "info", "mdFiles: " + str( fileCount ) )
  # want front matter to be before 01.md and back matter to be after 50.md
    ifCopy( os.path.join( dir, '_front', 'front-matter.md' ), 
            os.path.join( mfd, '00_front-matter.md' ))
    ifCopy( os.path.join( dir, '_back',  'back-matter.md'  ), 
            os.path.join( mfd, '51_back-matter.md' ))
Exemplo n.º 2
0
def write_json(outfile, p):
    """
    Simple wrapper to write a file as JSON.
    """
    make_dir(outfile.rsplit('/', 1)[0])
    f = codecs.open(outfile, 'w', encoding='utf-8')
    f.write(get_dump(p))
    f.close()
 def unzip_resource_only(self, zip_name, test_only):
     unpack_folder = self.unzip_resource(zip_name)
     out_dir = os.path.join(self.temp_dir, 'test_folder')
     file_utils.make_dir(out_dir)
     for f in test_only:
         shutil.copy(os.path.join(unpack_folder, f), out_dir)
     shutil.rmtree(unpack_folder, ignore_errors=True)
     return out_dir
Exemplo n.º 4
0
def create_pdf(lang_code, checking_level, version):
    global download_dir, unfoldingWord_dir

    # Create PDF via ConTeXt
    try:
        print_ok('BEGINNING: ', 'PDF generation.')
        out_dir = os.path.join(download_dir, 'make_pdf')
        make_dir(out_dir)

        # generate a tex file
        print('Generating tex file...', end=' ')
        tex_file = os.path.join(out_dir, '{0}.tex'.format(lang_code))

        # make sure it doesn't already exist
        if os.path.isfile(tex_file):
            os.remove(tex_file)

        with OBSTexExport(lang_code, tex_file, 0, '360px',
                          checking_level) as tex:
            tex.run()
        print('finished.')

        # run context
        print_notice('Running context - this may take several minutes.')

        # noinspection PyTypeChecker
        trackers = ','.join([
            'afm.loading', 'fonts.missing', 'fonts.warnings', 'fonts.names',
            'fonts.specifications', 'fonts.scaling', 'system.dump'
        ])

        cmd = 'context --paranoid --batchmode --trackers={0} "{1}"'.format(
            trackers, tex_file)

        try:
            subprocess.check_call(cmd,
                                  shell=True,
                                  stderr=subprocess.STDOUT,
                                  cwd=out_dir)

        except subprocess.CalledProcessError as e:
            if e.message:
                raise e
        print('Finished running context.')

        print('Copying PDF to API...', end=' ')
        version = version.replace('.', '_')
        if version[0:1] != 'v':
            version = 'v' + version

        pdf_file = os.path.join(unfoldingWord_dir, lang_code,
                                'obs-{0}-{1}.pdf'.format(lang_code, version))
        shutil.copyfile(os.path.join(out_dir, '{0}.pdf'.format(lang_code)),
                        pdf_file)
        print('finished.')

    finally:
        print_ok('FINISHED:', 'generating PDF.')
def create_pdf(lang_code, checking_level, version):
    global download_dir, unfoldingWord_dir

    # Create PDF via ConTeXt
    try:
        print_ok('BEGINNING: ', 'PDF generation.')
        out_dir = os.path.join(download_dir, 'make_pdf')
        make_dir(out_dir)

        # generate a tex file
        print('Generating tex file...', end=' ')
        tex_file = os.path.join(out_dir, '{0}.tex'.format(lang_code))

        # make sure it doesn't already exist
        if os.path.isfile(tex_file):
            os.remove(tex_file)

        with OBSTexExport(lang_code, tex_file, 0, '360px', checking_level) as tex:
            tex.run()
        print('finished.')

        # run context
        print_notice('Running context - this may take several minutes.')

        # noinspection PyTypeChecker
        trackers = ','.join(['afm.loading', 'fonts.missing', 'fonts.warnings', 'fonts.names',
                             'fonts.specifications', 'fonts.scaling', 'system.dump'])

        cmd = 'context --paranoid --batchmode --trackers={0} "{1}"'.format(trackers, tex_file)

        try:
            subprocess.check_call(cmd, shell=True, stderr=subprocess.STDOUT, cwd=out_dir)

        except subprocess.CalledProcessError as e:
            if e.message:
                raise e
        print('Finished running context.')

        print('Copying PDF to API...', end=' ')
        version = version.replace('.', '_')
        if version[0:1] != 'v':
            version = 'v' + version

        pdf_file = os.path.join(unfoldingWord_dir, lang_code, 'obs-{0}-{1}.pdf'.format(lang_code, version))
        shutil.copyfile(os.path.join(out_dir, '{0}.pdf'.format(lang_code)), pdf_file)
        print('finished.')

    finally:
        print_ok('FINISHED:', 'generating PDF.')
Exemplo n.º 6
0
    def run(self):

        if 'git.door43.org' not in self.source_repo_url:
            print_warning(
                'Currently only git.door43.org repositories are supported.')
            sys.exit(0)

        try:
            # clean up the git repo url
            if self.source_repo_url[-4:] == '.git':
                self.source_repo_url = self.source_repo_url[:-4]

            if self.source_repo_url[-1:] == '/':
                self.source_repo_url = self.source_repo_url[:-1]

            # download the archive
            file_to_download = join_url_parts(self.source_repo_url,
                                              'archive/master.zip')
            repo_dir = self.source_repo_url.rpartition('/')[2]
            downloaded_file = os.path.join(self.temp_dir, repo_dir + '.zip')
            try:
                if not self.quiet:
                    print('Downloading {0}...'.format(file_to_download),
                          end=' ')
                if not os.path.isfile(downloaded_file):
                    download_file(file_to_download, downloaded_file)
            finally:
                if not self.quiet:
                    print('finished.')

            # unzip the archive
            try:
                if not self.quiet:
                    print('Unzipping...'.format(downloaded_file), end=' ')
                unzip(downloaded_file, self.temp_dir)
            finally:
                if not self.quiet:
                    print('finished.')

            # get the manifest
            try:
                if not self.quiet:
                    print('Reading the manifest...', end=' ')
                manifest = load_json_object(
                    os.path.join(self.temp_dir, 'manifest.json'))
            finally:
                if not self.quiet:
                    print('finished.')

            # create output directory
            make_dir(self.output_directory)

            # read the markdown files and output html files
            try:
                if not self.quiet:
                    print('Processing the OBS markdown files')
                files_to_process = []
                for i in range(1, 51):
                    files_to_process.append(str(i).zfill(2) + '.md')

                current_dir = os.path.dirname(inspect.stack()[0][1])
                with codecs.open(os.path.join(current_dir, 'template.html'),
                                 'r', 'utf-8-sig') as html_file:
                    html_template = html_file.read()

                for file_to_process in files_to_process:

                    # read the markdown file
                    file_name = os.path.join(self.temp_dir, repo_dir,
                                             'content', file_to_process)
                    with codecs.open(file_name, 'r', 'utf-8-sig') as md_file:
                        md = md_file.read()

                    html = markdown.markdown(md)
                    html = TransformOBS.dir_re.sub(r'\1\n' + html + r'\n\2',
                                                   html_template)
                    write_file(
                        os.path.join(self.output_directory,
                                     file_to_process.replace('.md', '.html')),
                        html)

            except IOError as ioe:
                print_error('{0}: {1}'.format(ioe.strerror, ioe.filename))
                self.errors.append(ioe)

            except Exception as e:
                print_error(e.message)
                self.errors.append(e)

            finally:
                if not self.quiet:
                    print('finished.')

        except Exception as e:
            print_error(e.message)
            self.errors.append(e)
Exemplo n.º 7
0
 def test_make_dir(self):
     tmp_dir = tempfile.mkdtemp()
     sub_dir = tmp_dir + "/subdirectory"
     file_utils.make_dir(sub_dir)
     self.assertTrue(os.path.isdir(sub_dir))
Exemplo n.º 8
0
def main(git_repo, tag):
    global download_dir

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir, git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/' + tag + '.zip')
    metadata_obj = None
    content_dir = None
    toc_obj = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'meta.yaml' in files:
            # read the metadata
            try:
                print('Reading the metadata...', end=' ')
                metadata_obj = TAMetaData(os.path.join(root, 'meta.yaml'))
            finally:
                print('finished.')

        if 'toc.yaml' in files:
            # read the table of contents
            try:
                print('Reading the toc...', end=' ')
                toc_obj = TATableOfContents(os.path.join(root, 'toc.yaml'))
            finally:
                print('finished.')

        if 'content' in dirs:
            content_dir = os.path.join(root, 'content')

        # if we have everything, exit the loop
        if content_dir and metadata_obj and toc_obj:
            break

    # check for valid repository structure
    if not metadata_obj:
        print_error('Did not find meta.yaml in {}'.format(git_repo))
        sys.exit(1)

    if not content_dir:
        print_error('Did not find the content directory in {}'.format(git_repo))
        sys.exit(1)

    if not toc_obj:
        print_error('Did not find toc.yaml in {}'.format(git_repo))
        sys.exit(1)

    # check for missing pages
    check_missing_pages(toc_obj, content_dir)

    # generate the pages
    print('Generating the manual...', end=' ')
    manual = TAManual(metadata_obj, toc_obj)
    manual.load_pages(content_dir)
    print('finished.')

    file_name = os.path.join(get_output_dir(), '{0}_{1}.json'.format(manual.meta.manual, manual.meta.volume))
    print('saving to {0} ...'.format(file_name), end=' ')
    content = json.dumps(manual, sort_keys=True, indent=2, cls=TAEncoder)
    write_file(file_name, content)
    print('finished.')
Exemplo n.º 9
0
 def test_make_dir(self):
     self.tmp_dir = tempfile.mkdtemp(prefix='Door43_test_file_utils_')
     sub_dir = os.path.join(self.tmp_dir, 'subdirectory')
     file_utils.make_dir(sub_dir)
     self.assertTrue(os.path.isdir(sub_dir))
def main(git_repo, tag, no_pdf):
    global download_dir

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    today = ''.join(str(datetime.date.today()).rsplit('-')[0:3])  # str(datetime.date.today())
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir, git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/' + tag + '.zip')
    manifest = None
    status = None  # type: OBSStatus
    content_dir = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'manifest.json' in files:
            # read the manifest
            try:
                print('Reading the manifest...', end=' ')
                content_dir = root
                manifest = load_json_object(os.path.join(root, 'manifest.json'))
            finally:
                print('finished.')

        if 'status.json' in files:
            # read the meta data
            try:
                print('Reading the status...', end=' ')
                content_dir = root
                status = OBSStatus(os.path.join(root, 'status.json'))
            finally:
                print('finished.')

        # if we have everything, exit the loop
        if content_dir and manifest and status:
            break

    # check for valid repository structure
    if not manifest:
        print_error('Did not find manifest.json in {}'.format(git_repo))
        sys.exit(1)

    if not status:
        print_error('Did not find status.json in {}'.format(git_repo))
        sys.exit(1)

    print('Initializing OBS object...', end=' ')
    lang = manifest['target_language']['id']
    obs_obj = OBS()
    obs_obj.date_modified = today
    obs_obj.direction = manifest['target_language']['direction']
    obs_obj.language = lang
    print('finished')

    obs_obj.chapters = load_obs_chapters(content_dir)
    obs_obj.chapters.sort(key=lambda c: c['number'])

    if not obs_obj.verify_all():
        print_error('Quality check did not pass.')
        sys.exit(1)

    print('Loading languages...', end=' ')
    lang_dict = OBS.load_lang_strings()
    print('finished.')

    print('Loading the catalog...', end=' ')
    export_dir = '/var/www/vhosts/door43.org/httpdocs/exports'
    # uw_cat_path = os.path.join(unfoldingWord_dir, 'obs-catalog.json')
    # uw_catalog = load_json_object(uw_cat_path, [])
    # uw_cat_langs = [x['language'] for x in uw_catalog]
    cat_path = os.path.join(export_dir, 'obs-catalog.json')
    catalog = load_json_object(cat_path, [])
    print('finished')

    print('Getting already published languages...', end=' ')
    json_lang_file_path = os.path.join(export_dir, lang, 'obs', 'obs-{0}.json'.format(lang))
    # prev_json_lang = load_json_object(json_lang_file_path, {})

    if lang not in lang_dict:
        print("Configuration for language {0} missing.".format(lang))
        sys.exit(1)
    print('finished.')

    updated = update_language_catalog(lang, obs_obj.direction, status, today, lang_dict, catalog)

    print('Writing the OBS file to the exports directory...', end=' ')
    cur_json = json.dumps(obs_obj, sort_keys=True, cls=OBSEncoder)

    if updated:
        ([x for x in catalog if x['language'] == lang][0]['date_modified']) = today
        write_file(json_lang_file_path.replace('.txt', '.json'), cur_json)
    print('finished.')

    export_to_api(lang, status, today, cur_json)

    cat_json = json.dumps(catalog, sort_keys=True, cls=OBSEncoder)
    write_file(cat_path, cat_json)

    # update the catalog
    print_ok('STARTING: ', 'updating the catalogs.')
    update_catalog()
    print_ok('FINISHED: ', 'updating the catalogs.')

    if no_pdf:
        return

    create_pdf(lang, status.checking_level, status.version)
def main(git_repo, tag, domain):
    global download_dir, out_template

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    today = ''.join(str(datetime.date.today()).rsplit('-')[0:3])  # str(datetime.date.today())
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir, git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/' + tag + '.zip')
    manifest = None
    metadata_obj = None
    content_dir = ''
    usfm_file = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'manifest.json' in files:
            # read the manifest
            try:
                print('Reading the manifest...', end=' ')
                manifest = load_json_object(os.path.join(root, 'manifest.json'))
                content_dir = root

                # look for the usfm file for the whole book
                found_usfm = glob(os.path.join(content_dir, '*.usfm'))
                if len(found_usfm) == 1:
                    usfm_file = os.path.join(content_dir, found_usfm[0])
            finally:
                print('finished.')

        if 'meta.json' in files:
            # read the metadata
            try:
                print('Reading the metadata...', end=' ')
                metadata_obj = BibleMetaData(os.path.join(root, 'meta.json'))
            finally:
                print('finished.')

        # if we have everything, exit the loop
        if manifest and metadata_obj:
            break

    # check for valid repository structure
    if not manifest:
        print_error('Did not find manifest.json in {}'.format(git_repo))
        sys.exit(1)

    if not metadata_obj:
        print_error('Did not find meta.json in {}'.format(git_repo))
        sys.exit(1)

    # get the versification data
    print('Getting versification info...', end=' ')
    vrs = Bible.get_versification(metadata_obj.versification)  # type: list<Book>

    # get the book object for this repository
    book = next((b for b in vrs if b.book_id.lower() == manifest['project']['id']), None)  # type: Book
    if not book:
        print_error('Book versification data was not found for "{}"'.format(manifest['project']['id']))
        sys.exit(1)
    print('finished')

    if usfm_file:
        read_unified_file(book, usfm_file)

    else:
        read_chunked_files(book, content_dir, metadata_obj)

    # do basic checks
    print('Running USFM checks...', end=' ')
    book.verify_chapters_and_verses(True)
    if book.validation_errors:
        print_error('These USFM errors must be corrected before publishing can continue.')
        sys.exit(1)
    else:
        print('finished.')

    # insert paragraph markers
    print('Inserting paragraph markers...', end=' ')
    Bible.insert_paragraph_markers(book)
    print('finished.')

    # get chunks for this book
    print('Chunking the text...', end=' ')
    Bible.chunk_book(metadata_obj.versification, book)
    book.apply_chunks()
    print('finished.')

    # save the output
    out_dir = out_template.format(domain, metadata_obj.slug)

    # produces something like '01-GEN.usfm'
    book_file_name = '{0}-{1}.usfm'.format(str(book.number).zfill(2), book.book_id)
    print('Writing ' + book_file_name + '...', end=' ')
    write_file('{0}/{1}'.format(out_dir, book_file_name), book.usfm)
    print('finished.')

    # look for an existing status.json file
    print('Updating the status for {0}...'.format(metadata_obj.lang), end=' ')
    status_file = '{0}/status.json'.format(out_dir)
    if os.path.isfile(status_file):
        status = BibleStatus(status_file)
    else:
        status = BibleStatus()

    status.update_from_meta_data(metadata_obj)

    # add this book to the list of "books_published"
    status.add_book_published(book)

    # update the "date_modified"
    status.date_modified = today
    print('finished.')

    # save the status.json file
    print('Writing status.json...', end=' ')
    status_json = json.dumps(status, sort_keys=True, indent=2, cls=BibleEncoder)
    write_file(status_file, status_json)
    print('finished')

    # let the API know it is there
    print('Publishing to the API...')
    with api_publish(out_dir) as api:
        api.run()
    print('Finished publishing to the API.')

    # update the catalog
    print()
    print('Updating the catalogs...', end=' ')
    update_catalog()
    print('finished.')

    print_notice('Check {0} and do a git push'.format(out_dir))
Exemplo n.º 12
0
import common

body_re = re.compile(r'<body.*?>(.*)</body>', re.UNICODE | re.DOTALL)
verse_re = re.compile(r'<a name=\'\d{4}(\d{2})\'>.*?<table>(.*?)</table>',
                      re.UNICODE | re.DOTALL)
tags_ai_re = re.compile(r'(?:<a.*?>|<i.*?>)(.*?)(?:</a.*?>|</i.*?>)',
                        re.UNICODE)
spans_re = re.compile(r'</?span.*?>', re.UNICODE)
td_re = re.compile(r'(<td)(.*?)(>)', re.UNICODE)
values_re = re.compile(r'<td>(.*?)</td>', re.UNICODE)

if __name__ == '__main__':
    nt_books = common.get_versification()  # type: list<dict>
    this_dir = os.path.dirname(inspect.stack()[0][1])
    out_dir = os.path.join(this_dir, 'OutFiles')
    make_dir(out_dir)

    all_file_lines = [
        '"book_num","book_id","chapter","verse","manuscript","date","words"'
    ]

    for nt_book in nt_books:  # type: dict
        file_lines = [
            '"book_num","book_id","chapter","verse","manuscript","date","words"'
        ]
        book_id = nt_book['idx'] - 40
        book_id_str = str(book_id).zfill(2)
        url = 'http://greek-language.com/cntr/collation/{0}{1}.htm'

        for nt_chapter in nt_book['chapters']:
            print('Getting {0} {1}.'.format(nt_book['id'], nt_chapter[0]))
Exemplo n.º 13
0
def import_obs(lang_data, git_repo, door43_url, no_pdf):
    global download_dir, root, api_dir

    lang_code = lang_data['lc']

    # pre-flight checklist
    link_source = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/jpg/1/en'
    if not os.path.isdir(link_source):
        print_error(
            'Image source directory not found: {0}.'.format(link_source))
        sys.exit(1)

    if git_repo[-1:] != '/':
        git_repo += '/'

    if no_pdf:
        tools_dir = None
    else:
        tools_dir = '/var/www/vhosts/door43.org/tools'
        if not os.path.isdir(tools_dir):
            tools_dir = os.path.expanduser('~/Projects/tools')

        # prompt if tools not found
        if not os.path.isdir(tools_dir):
            tools_dir = None
            print_notice(
                'The tools directory was not found. The PDF cannot be generated.'
            )
            resp = prompt(
                'Do you want to continue without generating a PDF? [Y|n]: ')
            if resp != '' and resp != 'Y' and resp != 'y':
                sys.exit(0)

    if git_repo:
        if git_repo[-1:] == '/':
            git_repo = git_repo[:-1]

        download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
        make_dir(download_dir)

        # make sure OBS is initialized on Dokuwiki
        test_dir = root.format(lang_code, '')
        if not os.path.isdir(test_dir):
            print_warning(
                'It seems OBS has not been initialized on Door43.org for {0}'.
                format(lang_code))
            sys.exit(1)

    elif door43_url:
        print_error('URL not yet implemented.')
        return
    else:
        print_error('Source not provided.')
        return

    # get the source files from the git repository
    if 'github' in git_repo:
        # https://github.com/unfoldingWord/obs-ru
        # https://raw.githubusercontent.com/unfoldingWord/obs-ru/master/obs-ru.json
        raw_url = git_repo.replace('github.com', 'raw.githubusercontent.com')
    elif 'git.door43.org' in git_repo:
        raw_url = join_url_parts(git_repo, 'raw')
    else:
        # this is to keep IntelliJ happy, is should have been caught in sub main
        return

    # download needed files from the repository
    file_suffix = '-{0}.json'.format(lang_code.lower())
    files_to_download = [
        join_url_parts(raw_url, 'master/obs' + file_suffix),
        join_url_parts(raw_url, 'master/status' + file_suffix)
    ]

    for file_to_download in files_to_download:

        downloaded_file = os.path.join(download_dir,
                                       file_to_download.rpartition('/')[2])

        try:
            print('Downloading {0}...'.format(file_to_download), end=' ')
            download_file(file_to_download, downloaded_file)
        finally:
            print('finished.')

    # read the files from the git repository
    file_suffix = '-{0}.json'.format(lang_code.lower())
    obs_obj = None
    status_obj = None
    # front_matter_found = False
    # back_matter_found = False

    try:
        print('Examining the files...', end=' ')
        for root_path, dirs, files in os.walk(download_dir):

            if not len(files):
                continue

            for git_file in files:
                if git_file == 'obs' + file_suffix:
                    obs_obj = OBS(os.path.join(root_path, git_file))
                elif git_file == 'status' + file_suffix:
                    status_obj = OBSStatus(os.path.join(root_path, git_file))
                    # elif 'front-matter' in git_file:
                    #     front_matter_found = True
                    # elif 'back-matter' in git_file:
                    #     back_matter_found = True
    finally:
        print('finished.')

    # check data integrity
    if not obs_obj.verify_all():
        sys.exit(1)

    if not status_obj:
        print_error(
            'The file "status{0}" was not found in the git repository.'.format(
                file_suffix))
        sys.exit(1)

    # create Dokuwiki pages
    print_ok('Begin: ', 'creating Dokuwiki pages.')
    for chapter in obs_obj.chapters:

        chapter_title = '====== {0} ======'.format(chapter['title'])
        chapter_ref = '//{0}//'.format(chapter['ref'])
        chapter_body = ''
        chapter_num = chapter['number'].zfill(2)

        for frame in chapter['frames']:
            chapter_body += '{{{{{0}?direct&}}}}\n\n{1}\n\n'.format(
                frame['img'], frame['text'])

        file_name = root.format(lang_code, chapter_num + '.txt')
        print('  Writing {0}'.format(file_name))
        with codecs.open(file_name, 'w', 'utf-8-sig') as out_file:
            out_file.write('{0}\n\n{1}{2}\n'.format(chapter_title,
                                                    chapter_body, chapter_ref))

    print_ok('Finished: ', 'creating Dokuwiki pages.')

    # Create image symlinks on api.unfoldingword.org
    try:
        print('Creating symlink to images directory...', end=' ')
        link_name = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/jpg/1/{0}'.format(
            lang_code.lower())
        if not os.path.isfile(link_name) and not os.path.isdir(
                link_name) and not os.path.islink(link_name):
            os.symlink(link_source, link_name)
    finally:
        print('finished.')

    # Create PDF via ConTeXt
    if not no_pdf and tools_dir and os.path.isdir(tools_dir):
        try:
            print_ok('Beginning: ', 'PDF generation.')
            script_file = os.path.join(tools_dir, 'obs/book/pdf_export.sh')
            out_dir = api_dir.format(lang_code)
            make_dir(out_dir)
            process = subprocess.Popen([
                script_file, '-l', lang_code, '-c', status_obj.checking_level,
                '-v', status_obj.version, '-o', out_dir
            ],
                                       shell=True,
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE)

            # wait for the process to terminate
            out, err = process.communicate()
            exit_code = process.returncode
            out = out.strip().decode('utf-8')
            err = err.strip().decode('utf-8')

            # the error message may be in stdout
            if exit_code != 0:
                if not err:
                    err = out
                    out = None

            if err:
                print_error(err, 2)

            if out:
                print('  ' + out)

            print('  PDF subprocess finished with exit code {0}'.format(
                exit_code))

        finally:
            print_ok('Finished:', 'generating PDF.')
    try:
        print('Creating symlink to images directory...', end=' ')
        link_name = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/jpg/1/{0}'.format(lang.lower())
        if not os.path.isfile(link_name) and not os.path.isdir(link_name) and not os.path.islink(link_name):
            os.symlink(link_source, link_name)
    finally:
        print('finished.')

    # Create PDF via ConTeXt
    if not no_pdf and tools_dir and os.path.isdir(tools_dir):
        try:
            print_ok('Beginning: ', 'PDF generation.')
            script_file = os.path.join(tools_dir, 'obs/book/pdf_export.sh')
            api_dir = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/txt/1/{0}'
            out_dir = api_dir.format(lang)
            make_dir(out_dir)
            process = subprocess.Popen([script_file,
                                        '-l', lang,
                                        '-c', status['checking_level'],
                                        '-v', status['version'],
                                        '-o', out_dir],
                                       shell=True,
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE)

            # wait for the process to terminate
            out, err = process.communicate()
            exit_code = process.returncode
            out = out.strip().decode('utf-8')
            err = err.strip().decode('utf-8')
Exemplo n.º 15
0
    def run(self):
        # download the archive
        file_to_download = self.source_url
        filename = self.source_url.rpartition('/')[2]
        downloaded_file = os.path.join(self.download_dir, filename)
        self.log_message('Downloading {0}...'.format(file_to_download))
        if not os.path.isfile(downloaded_file):
            try:
                download_file(file_to_download, downloaded_file)
            finally:
                if not os.path.isfile(downloaded_file):
                    raise Exception("Failed to download {0}".format(file_to_download))
                else:
                    self.log_message('Download successful.')

        # unzip the archive
        self.log_message('Unzipping {0}...'.format(downloaded_file))
        unzip(downloaded_file, self.files_dir)
        self.log_message('Unzip successful.')

        # create output directory
        make_dir(self.output_dir)

        # read the markdown files and output html files
        self.log_message('Processing the OBS markdown files')

        files = sorted(glob(os.path.join(self.files_dir, '*')))

        current_dir = os.path.dirname(os.path.realpath(__file__))
        with open(os.path.join(current_dir, 'obs-template.html')) as template_file:
            html_template = string.Template(template_file.read())

        complete_html = ''
        for filename in files:
            if filename.endswith('.md'):
                # read the markdown file
                with codecs.open(filename, 'r', 'utf-8-sig') as md_file:
                    md = md_file.read()
                html = markdown.markdown(md)
                complete_html += html
                html = html_template.safe_substitute(content=html)
                html_filename = os.path.splitext(os.path.basename(filename))[0] + ".html"
                output_file = os.path.join(self.output_dir, html_filename)
                write_file(output_file, html)
                self.log_message('Converted {0} to {1}.'.format(os.path.basename(filename), os.path.basename(html_filename)))
            else:
                try:
                    output_file = os.path.join(self.output_dir, filename[len(self.files_dir)+1:])
                    if not os.path.exists(output_file):
                        if not os.path.exists(os.path.dirname(output_file)):
                            os.makedirs(os.path.dirname(output_file))
                        copyfile(filename, output_file)
                except Exception:
                    pass

        # Do the OBS inspection
        inspector = OBSInspection(self.output_dir)
        try:
            inspector.run()
        except Exception as e:
            self.warning_message('Failed to run OBS inspector: {0}'.format(e.message))

        for warning in inspector.warnings:
            self.warning_message(warning)
        for error in inspector.errors:
            self.error_message(error)

        complete_html = html_template.safe_substitute(content=complete_html)
        write_file(os.path.join(self.output_dir, 'all.html'), complete_html)

        self.log_message('Made one HTML of all stories in all.html.')
        self.log_message('Finished processing Markdown files.')
Exemplo n.º 16
0
def main(git_repo, tag, no_pdf):
    global download_dir

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    today = ''.join(str(datetime.date.today()).rsplit(
        str('-'))[0:3])  # str(datetime.date.today())
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir,
                                           git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/{0}.zip'.format(tag))
    manifest = None
    status = None  # type: OBSStatus
    content_dir = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'manifest.json' in files:
            # read the manifest
            try:
                print('Reading the manifest...', end=' ')
                content_dir = root
                manifest = load_json_object(os.path.join(
                    root, 'manifest.json'))
                status = OBSStatus.from_manifest(manifest)
            finally:
                print('finished.')

        if 'content' in dirs:
            content_dir = os.path.join(root, 'content')

        # if we have everything, exit the loop
        if content_dir and manifest and status:
            break

    # check for valid repository structure
    if not manifest:
        print_error('Did not find manifest.json in {}'.format(git_repo))
        sys.exit(1)

    print('Initializing OBS object...', end=' ')
    lang = manifest['language']['slug']
    obs_obj = OBS()
    obs_obj.date_modified = today
    obs_obj.direction = manifest['language']['dir']
    obs_obj.language = lang
    print('finished')

    obs_obj.chapters = load_obs_chapters(content_dir)
    obs_obj.chapters.sort(key=lambda c: int(c['number']))

    if not obs_obj.verify_all():
        print_error('Quality check did not pass.')
        sys.exit(1)

    print('Loading languages...', end=' ')
    lang_dict = OBS.load_lang_strings()
    print('finished.')

    print('Loading the catalog...', end=' ')
    export_dir = '/var/www/vhosts/door43.org/httpdocs/exports'

    cat_path = os.path.join(export_dir, 'obs-catalog.json')
    catalog = load_json_object(cat_path, [])
    print('finished')

    print('Getting already published languages...', end=' ')
    json_lang_file_path = os.path.join(export_dir, lang, 'obs',
                                       'obs-{0}.json'.format(lang))

    if lang not in lang_dict:
        print("Configuration for language {0} missing.".format(lang))
        sys.exit(1)
    print('finished.')

    updated = update_language_catalog(lang, obs_obj.direction, status, today,
                                      lang_dict, catalog)

    print('Writing the OBS file to the exports directory...', end=' ')
    cur_json = json.dumps(obs_obj, sort_keys=True, cls=OBSEncoder)

    if updated:
        ([x for x in catalog
          if x['language'] == lang][0]['date_modified']) = today
        # noinspection PyTypeChecker
        write_file(json_lang_file_path.replace('.txt', '.json'), cur_json)
    print('finished.')

    export_to_api(lang, status, today, cur_json)

    cat_json = json.dumps(catalog, sort_keys=True, cls=OBSEncoder)
    write_file(cat_path, cat_json)

    # update the catalog
    print_ok('STARTING: ', 'updating the catalogs.')
    update_catalog()
    print_ok('FINISHED: ', 'updating the catalogs.')

    if no_pdf:
        return

    create_pdf(lang, status.checking_level, status.version)
Exemplo n.º 17
0
def handle(event, context):
    data = event['data']

    commit_id = data['after']
    commit = None
    for commit in data['commits']:
        if commit['id'] == commit_id:
            break

    commit_url = commit['url']
    commit_message = commit['message']

    if 'https://git.door43.org/' not in commit_url and 'http://test.door43.org:3000/' not in commit_url:
        raise Exception('Currently only git.door43.org repositories are supported.')

    pre_convert_bucket = event['pre_convert_bucket']
    cdn_bucket = event['cdn_bucket']
    gogs_user_token = event['gogs_user_token']
    api_url = event['api_url']
    repo_name = data['repository']['name']
    repo_owner = data['repository']['owner']['username']
    compare_url = data['compare_url']

    if 'pusher' in data:
        pusher = data['pusher']
    else:
        pusher = {'username': commit['author']['username']}
    pusher_username = pusher['username']

    # The following sections of code will:
    # 1) download and unzip repo files
    # 2) massage the repo files by creating a new directory and file structure
    # 3) zip up the massages filed
    # 4) upload massaged files to S3 in zip file

    # 1) Download and unzip the repo files
    repo_zip_url = commit_url.replace('commit', 'archive') + '.zip'
    repo_zip_file = os.path.join(tempfile.gettempdir(), repo_zip_url.rpartition('/')[2])
    repo_dir = tempfile.mkdtemp(prefix='repo_')
    try:
        print('Downloading {0}...'.format(repo_zip_url), end=' ')
        if not os.path.isfile(repo_zip_file):
            download_file(repo_zip_url, repo_zip_file)
    finally:
        print('finished.')

    # Unzip the archive
    try:
        print('Unzipping {0}...'.format(repo_zip_file), end=' ')
        unzip(repo_zip_file, repo_dir)
    finally:
        print('finished.')

    # 2) Massage the content to just be a directory of MD files in alphabetical order as they should be compiled together in the converter
    content_dir = os.path.join(repo_dir, repo_name, 'content')
    md_files = glob(os.path.join(content_dir, '*.md'))
    massaged_files_dir = tempfile.mktemp(prefix='files_')
    make_dir(massaged_files_dir)
    print('Massaging content from {0} to {1}...'.format(content_dir, massaged_files_dir), end=' ')
    for md_file in md_files:
        copyfile(md_file, os.path.join(massaged_files_dir, os.path.basename(md_file)))
    # want front matter to be before 01.md and back matter to be after 50.md
    copyfile(os.path.join(content_dir, '_front', 'front-matter.md'), os.path.join(massaged_files_dir, '00_front-matter.md'))
    copyfile(os.path.join(content_dir, '_back', 'back-matter.md'), os.path.join(massaged_files_dir, '51_back-matter.md'))
    print('finished.')

    # 3) Zip up the massaged files
    zip_filename = context.aws_request_id+'.zip' # context.aws_request_id is a unique ID for this lambda call, so using it to not conflict with other requests
    zip_filepath = os.path.join(tempfile.gettempdir(), zip_filename)
    md_files = glob(os.path.join(massaged_files_dir, '*.md'))
    print('Zipping files from {0} to {1}...'.format(massaged_files_dir, zip_filepath), end=' ')
    for md_file in md_files:
        add_file_to_zip(zip_filepath, md_file, os.path.basename(md_file))
    print('finished.')

    # 4) Upload zipped file to the S3 bucket (you may want to do some try/catch and give an error if fails back to Gogs)
    print('Uploading {0} to {1}...'.format(zip_filepath, pre_convert_bucket), end=' ')
    s3_client = boto3.client('s3')
    s3_client.upload_file(zip_filepath, pre_convert_bucket, zip_filename)
    print('finished.')

    # Send job request to tx-manager
    source_url = 'https://s3-us-west-2.amazonaws.com/'+pre_convert_bucket+'/'+zip_filename # we use us-west-2 for our s3 buckets
    tx_manager_job_url = api_url+'/tx/job'
    payload = {
        "identifier": "{0}:::{1}:::{2}".format(repo_owner, repo_name, commit_id),
        "user_token": gogs_user_token,
        "username": pusher_username,
        "resource_type": "obs",
        "input_format": "md",
        "output_format": "html",
        "source": source_url,
        "callback": api_url+'/sampleclient/callback'
    }
    headers = {"content-type": "application/json"}

    print('Making request to tx-Manager URL {0} with payload:'.format(tx_manager_job_url))
    print(payload)
    print('...', end=' ')
    response = requests.post(tx_manager_job_url, json=payload, headers=headers)
    print('finished.')

    # for testing
    print('tx-manager response:')
    print(response)
    json_data = json.loads(response.text)
    print("json:")
    print(json_data)

    build_log_json = {
        'job_id': json_data['job']['job_id'],
        'repo_name': repo_name,
        'repo_owner': repo_owner,
        'commit_id': commit_id,
        'committed_by': pusher_username,
        'commit_url': commit_url,
        'compare_url': compare_url,
        'commit_message': commit_message,
        'request_timestamp': datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
        'eta_timestamp': json_data['job']['eta'],
        'success': None,
        'status': 'started',
        'message': 'Conversion in progress...'
    }

    if 'errorMessage' in json_data:
        build_log_json['status'] = 'failed'
        build_log_json['message'] = json_data['errorMessage']

    # Make a build_log.json file with this repo and commit data for later processing, upload to S3
    s3_project_key = 'u/{0}/{1}/{2}'.format(repo_owner, repo_name, commit_id[:10])
    s3_resource = boto3.resource('s3')
    bucket = s3_resource.Bucket(cdn_bucket)
    for obj in bucket.objects.filter(Prefix=s3_project_key):
        s3_resource.Object(bucket.name, obj.key).delete()
    build_log_file = os.path.join(tempfile.gettempdir(), 'build_log_request.json')
    write_file(build_log_file, build_log_json)
    bucket.upload_file(build_log_file, s3_project_key+'/build_log.json', ExtraArgs={'ContentType': 'application/json'})
    print('Uploaded the following content from {0} to {1}/build_log.json'.format(build_log_file, s3_project_key))
    print(build_log_json)

    # If there was an error, in order to trigger a 400 error in the API Gateway, we need to raise an
    # exception with the returned 'errorMessage' because the API Gateway needs to see 'Bad Request:' in the string
    if 'errorMessage' in json_data:
        raise Exception(json_data['errorMessage'])

    return build_log_json
Exemplo n.º 18
0
    def run(self):

        try:
            self.temp_dir = tempfile.mkdtemp(prefix='txOBS_')

            # clean up the git repo url
            if self.source_repo_url[-4:] == '.git':
                self.source_repo_url = self.source_repo_url[:-4]

            if self.source_repo_url[-1:] == '/':
                self.source_repo_url = self.source_repo_url[:-1]

            # download the archive
            file_to_download = join_url_parts(self.source_repo_url, 'archive/master.zip')
            repo_dir = self.source_repo_url.rpartition('/')[2]
            downloaded_file = os.path.join(self.temp_dir, repo_dir + '.zip')
            try:
                print('Downloading {0}...'.format(file_to_download), end=' ')
                if not os.path.isfile(downloaded_file):
                    download_file(file_to_download, downloaded_file)
            finally:
                print('finished.')

            # unzip the archive
            try:
                print('Unzipping...'.format(downloaded_file), end=' ')
                unzip(downloaded_file, self.temp_dir)
            finally:
                print('finished.')

            # get the manifest
            try:
                print('Reading the manifest...', end=' ')
                manifest = load_json_object(os.path.join(self.temp_dir, 'manifest.json'))
            finally:
                print('finished.')

            # create output directory
            make_dir(self.output_directory)

            # read the markdown files and output html files
            try:
                print('Processing the OBS markdown files')
                files_to_process = []
                for i in range(1, 51):
                    files_to_process.append(str(i).zfill(2) + '.md')

                current_dir = os.path.dirname(inspect.stack()[0][1])
                with codecs.open(os.path.join(current_dir, 'template.html'), 'r', 'utf-8-sig') as html_file:
                    html_template = html_file.read()

                for file_to_process in files_to_process:

                    # read the markdown file
                    file_name = os.path.join(self.temp_dir, repo_dir, 'content', file_to_process)
                    with codecs.open(file_name, 'r', 'utf-8-sig') as md_file:
                        md = md_file.read()

                    html = markdown.markdown(md)
                    html = TransformOBS.dir_re.sub(r'\1\n' + html + r'\n\2', html_template)
                    write_file(os.path.join(self.output_directory, file_to_process.replace('.md', '.html')), html)

            except IOError as ioe:
                print_error('{0}: {1}'.format(ioe.strerror, ioe.filename))
                self.errors.append(ioe)

            except Exception as e:
                print_error(e.message)
                self.errors.append(e)

            finally:
                print('finished.')

        except Exception as e:
            print_error(e.message)
            self.errors.append(e)
Exemplo n.º 19
0
def main(git_repo, tag, domain):

    global download_dir, out_template

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    today = ''.join(str(datetime.date.today()).rsplit('-')[0:3])
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir, git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/' + tag + '.zip')
    books_published = {}
    metadata_obj = None
    usfm_dir = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'meta.json' in files:
            # read the metadata
            try:
                print('Reading the metadata...', end=' ')
                metadata_obj = BibleMetaData(os.path.join(root, 'meta.json'))
            finally:
                print('finished.')

        if 'usfm' in dirs:
            usfm_dir = os.path.join(root, 'usfm')

        # if we have everything, exit the loop
        if usfm_dir and metadata_obj:
            break

    # check for valid repository structure
    if not metadata_obj:
        print_error('Did not find meta.json in {}'.format(git_repo))
        sys.exit(1)

    if not usfm_dir:
        print_error('Did not find the usfm directory in {}'.format(git_repo))
        sys.exit(1)

    # get the versification data
    vrs = Bible.get_versification(metadata_obj.versification)  # type: list<Book>
    out_dir = out_template.format(domain, metadata_obj.slug, metadata_obj.lang)

    # walk through the usfm files
    usfm_files = glob(os.path.join(usfm_dir, '*.usfm'))
    errors_found = False
    for usfm_file in usfm_files:

        # read the file
        with codecs.open(usfm_file, 'r', 'utf-8') as in_file:
            book_text = in_file.read()

        # get the book id
        book_search = id_re.search(book_text)
        if not book_search:
            print_error('Book id not found in {}'.format(usfm_file))
            sys.exit(1)

        book_id = book_search.group(1)

        print('Beginning {}...'.format(book_id), end=' ')

        # get book versification info
        book = next((b for b in vrs if b.book_id == book_id), None)
        if not book:
            print_error('Book versification data was not found for "{}"'.format(book_id))
            sys.exit(1)

        # remove \s5 lines
        book_text = s5_re.sub('', book_text)

        # get the usfm for the book
        book.set_usfm(book_text)

        # do basic checks
        book.verify_usfm_tags()
        book.verify_chapters_and_verses(True)
        if book.validation_errors:
            errors_found = True

        # get chunks for this book
        Bible.chunk_book(metadata_obj.versification, book)
        book.apply_chunks()

        # produces something like '01-GEN.usfm'
        book_file_name = '{0}-{1}.usfm'.format(str(book.number).zfill(2), book.book_id)
        print('Writing ' + book_file_name + '...', end=' ')
        write_file('{0}/{1}'.format(out_dir, book_file_name), book.usfm)

        meta = ['Bible: OT']
        if book.number > 39:
            meta = ['Bible: NT']
        books_published[book.book_id.lower()] = {'name': book.name,
                                                 'meta': meta,
                                                 'sort': str(book.number).zfill(2),
                                                 'desc': ''
                                                 }
        print('finished.')

    # stop if errors were found
    if errors_found:
        print_error('These USFM errors must be corrected before publishing can continue.')
        sys.exit(1)

    print('Writing status.json...', end=' ')
    status = {"slug": '{0}'.format(metadata_obj.slug.lower()),
              "name": metadata_obj.name,
              "lang": metadata_obj.lang,
              "date_modified": today,
              "books_published": books_published,
              "status": {"checking_entity": metadata_obj.checking_entity,
                         "checking_level": metadata_obj.checking_level,
                         "comments": metadata_obj.comments,
                         "contributors": metadata_obj.contributors,
                         "publish_date": today,
                         "source_text": metadata_obj.source_text,
                         "source_text_version": metadata_obj.source_text_version,
                         "version": metadata_obj.version
                         }
              }
    write_file('{0}/status.json'.format(out_dir), status, indent=2)
    print('finished.')

    print()
    print('Publishing to the API...')
    with api_publish(out_dir) as api:
        api.run()
    print('Finished publishing to the API.')

    # update the catalog
    print()
    print('Updating the catalogs...', end=' ')
    update_catalog()
    print('finished.')

    print_notice('Check {0} and do a git push'.format(out_dir))
Exemplo n.º 20
0
def main(git_repo, tag):
    global download_dir

    # clean up the git repo url
    if git_repo[-4:] == '.git':
        git_repo = git_repo[:-4]

    if git_repo[-1:] == '/':
        git_repo = git_repo[:-1]

    # initialize some variables
    download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
    make_dir(download_dir)
    downloaded_file = '{0}/{1}.zip'.format(download_dir,
                                           git_repo.rpartition('/')[2])
    file_to_download = join_url_parts(git_repo, 'archive/' + tag + '.zip')
    metadata_obj = None
    content_dir = None
    toc_obj = None

    # download the repository
    try:
        print('Downloading {0}...'.format(file_to_download), end=' ')
        if not os.path.isfile(downloaded_file):
            download_file(file_to_download, downloaded_file)
    finally:
        print('finished.')

    try:
        print('Unzipping...'.format(downloaded_file), end=' ')
        unzip(downloaded_file, download_dir)
    finally:
        print('finished.')

    # examine the repository
    for root, dirs, files in os.walk(download_dir):

        if 'meta.yaml' in files:
            # read the metadata
            try:
                print('Reading the metadata...', end=' ')
                metadata_obj = TAMetaData(os.path.join(root, 'meta.yaml'))
            finally:
                print('finished.')

        if 'toc.yaml' in files:
            # read the table of contents
            try:
                print('Reading the toc...', end=' ')
                toc_obj = TATableOfContents(os.path.join(root, 'toc.yaml'))
            finally:
                print('finished.')

        if 'content' in dirs:
            content_dir = os.path.join(root, 'content')

        # if we have everything, exit the loop
        if content_dir and metadata_obj and toc_obj:
            break

    # check for valid repository structure
    if not metadata_obj:
        print_error('Did not find meta.yaml in {}'.format(git_repo))
        sys.exit(1)

    if not content_dir:
        print_error(
            'Did not find the content directory in {}'.format(git_repo))
        sys.exit(1)

    if not toc_obj:
        print_error('Did not find toc.yaml in {}'.format(git_repo))
        sys.exit(1)

    # check for missing pages
    check_missing_pages(toc_obj, content_dir)

    # generate the pages
    print('Generating the manual...', end=' ')
    manual = TAManual(metadata_obj, toc_obj)
    manual.load_pages(content_dir)
    print('finished.')

    file_name = os.path.join(
        get_output_dir(), '{0}_{1}.json'.format(manual.meta.manual,
                                                manual.meta.volume))
    print('saving to {0} ...'.format(file_name), end=' ')
    content = json.dumps(manual, sort_keys=True, indent=2, cls=TAEncoder)
    write_file(file_name, content)
    print('finished.')
Exemplo n.º 21
0
def import_obs(lang_data, git_repo, door43_url, no_pdf):
    global download_dir, root, api_dir

    lang_code = lang_data['lc']

    # pre-flight checklist
    link_source = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/jpg/1/en'
    if not os.path.isdir(link_source):
        print_error('Image source directory not found: {0}.'.format(link_source))
        sys.exit(1)

    if git_repo[-1:] != '/':
        git_repo += '/'

    if no_pdf:
        tools_dir = None
    else:
        tools_dir = '/var/www/vhosts/door43.org/tools'
        if not os.path.isdir(tools_dir):
            tools_dir = os.path.expanduser('~/Projects/tools')

        # prompt if tools not found
        if not os.path.isdir(tools_dir):
            tools_dir = None
            print_notice('The tools directory was not found. The PDF cannot be generated.')
            resp = prompt('Do you want to continue without generating a PDF? [Y|n]: ')
            if resp != '' and resp != 'Y' and resp != 'y':
                sys.exit(0)

    if git_repo:
        if git_repo[-1:] == '/':
            git_repo = git_repo[:-1]

        download_dir = '/tmp/{0}'.format(git_repo.rpartition('/')[2])
        make_dir(download_dir)

        # make sure OBS is initialized on Dokuwiki
        test_dir = root.format(lang_code, '')
        if not os.path.isdir(test_dir):
            print_warning('It seems OBS has not been initialized on Door43.org for {0}'.format(lang_code))
            sys.exit(1)

    elif door43_url:
        print_error('URL not yet implemented.')
        return
    else:
        print_error('Source not provided.')
        return

    # get the source files from the git repository
    if 'github' in git_repo:
        # https://github.com/unfoldingWord/obs-ru
        # https://raw.githubusercontent.com/unfoldingWord/obs-ru/master/obs-ru.json
        raw_url = git_repo.replace('github.com', 'raw.githubusercontent.com')
    elif 'git.door43.org' in git_repo:
        raw_url = join_url_parts(git_repo, 'raw')
    else:
        # this is to keep IntelliJ happy, is should have been caught in sub main
        return

    # download needed files from the repository
    file_suffix = '-{0}.json'.format(lang_code.lower())
    files_to_download = [
        join_url_parts(raw_url, 'master/obs' + file_suffix),
        join_url_parts(raw_url, 'master/status' + file_suffix)
    ]

    for file_to_download in files_to_download:

        downloaded_file = os.path.join(download_dir, file_to_download.rpartition('/')[2])

        try:
            print('Downloading {0}...'.format(file_to_download), end=' ')
            download_file(file_to_download, downloaded_file)
        finally:
            print('finished.')

    # read the files from the git repository
    file_suffix = '-{0}.json'.format(lang_code.lower())
    obs_obj = None
    status_obj = None
    # front_matter_found = False
    # back_matter_found = False

    try:
        print('Examining the files...', end=' ')
        for root_path, dirs, files in os.walk(download_dir):

            if not len(files):
                continue

            for git_file in files:
                if git_file == 'obs' + file_suffix:
                    obs_obj = OBS(os.path.join(root_path, git_file))
                elif git_file == 'status' + file_suffix:
                    status_obj = OBSStatus(os.path.join(root_path, git_file))
                    # elif 'front-matter' in git_file:
                    #     front_matter_found = True
                    # elif 'back-matter' in git_file:
                    #     back_matter_found = True
    finally:
        print('finished.')

    # check data integrity
    if not obs_obj.verify_all():
        sys.exit(1)

    if not status_obj:
        print_error('The file "status{0}" was not found in the git repository.'.format(file_suffix))
        sys.exit(1)

    # create Dokuwiki pages
    print_ok('Begin: ', 'creating Dokuwiki pages.')
    for chapter in obs_obj.chapters:

        chapter_title = '====== {0} ======'.format(chapter['title'])
        chapter_ref = '//{0}//'.format(chapter['ref'])
        chapter_body = ''
        chapter_num = chapter['number'].zfill(2)

        for frame in chapter['frames']:
            chapter_body += '{{{{{0}?direct&}}}}\n\n{1}\n\n'.format(frame['img'], frame['text'])

        file_name = root.format(lang_code, chapter_num + '.txt')
        print('  Writing {0}'.format(file_name))
        with codecs.open(file_name, 'w', 'utf-8-sig') as out_file:
            out_file.write('{0}\n\n{1}{2}\n'.format(chapter_title, chapter_body, chapter_ref))

    print_ok('Finished: ', 'creating Dokuwiki pages.')

    # Create image symlinks on api.unfoldingword.org
    try:
        print('Creating symlink to images directory...', end=' ')
        link_name = '/var/www/vhosts/api.unfoldingword.org/httpdocs/obs/jpg/1/{0}'.format(lang_code.lower())
        if not os.path.isfile(link_name) and not os.path.isdir(link_name) and not os.path.islink(link_name):
            os.symlink(link_source, link_name)
    finally:
        print('finished.')

    # Create PDF via ConTeXt
    if not no_pdf and tools_dir and os.path.isdir(tools_dir):
        try:
            print_ok('Beginning: ', 'PDF generation.')
            script_file = os.path.join(tools_dir, 'obs/book/pdf_export.sh')
            out_dir = api_dir.format(lang_code)
            make_dir(out_dir)
            process = subprocess.Popen([script_file,
                                        '-l', lang_code,
                                        '-c', status_obj.checking_level,
                                        '-v', status_obj.version,
                                        '-o', out_dir],
                                       shell=True,
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE)

            # wait for the process to terminate
            out, err = process.communicate()
            exit_code = process.returncode
            out = out.strip().decode('utf-8')
            err = err.strip().decode('utf-8')

            # the error message may be in stdout
            if exit_code != 0:
                if not err:
                    err = out
                    out = None

            if err:
                print_error(err, 2)

            if out:
                print('  ' + out)

            print('  PDF subprocess finished with exit code {0}'.format(exit_code))

        finally:
            print_ok('Finished:', 'generating PDF.')