コード例 #1
0
 def __generate_font__(self):
   gen_font1   = self.ui.rdoGenFont1.isChecked()
   temp_dir    = tempfile.mkdtemp(prefix = "sdse-")
   temp_name   = "font"
   font_type   = font_generator.FONT_TYPES.font01 if gen_font1 else font_generator.FONT_TYPES.font02
   for_game    = self.ui.chkGenForGame.isChecked()
   for_editor  = self.ui.chkGenForEditor.isChecked()
   
   font_data = self.get_font_data()
   if self.ui.rdoRightToLeft.isChecked():
     font_data.reverse()
   
   # Add the required characters to the end of the lowest-priority font
   # so they're there by default but they don't override any existing
   # settings for them, if they're already there.
   font_data[-1].chars += ''.join(REQUIRED_CHARS)
   
   font = font_generator.gen_font(font_data, font_type = font_type, img_width = 1024, draw_outlines = False)
   font.save(temp_dir, temp_name, for_game, for_editor, font_type, game = font_generator.GAMES.sdr2)
   
   basename  = os.path.join(temp_dir, temp_name)
   font_png  = basename + ".png"
   font_bmp  = basename + ".bmp"
   font_font = basename + ".font"
   
   font_dir  = os.path.join(common.editor_config.data01_dir, "jp", "font", "font.pak")
   
   if for_game:
     game_bmp  = "0000.bmp"  if gen_font1 else "0002.bmp"
     game_font = "0001.font" if gen_font1 else "0003.font"
     
     backup_files(font_dir, [game_bmp, game_font], suffix = "_FONT")
     # backup_time = time.strftime("%Y.%m.%d_%H.%M.%S_FONT")
     # backup_dir = os.path.join(common.editor_config.backup_dir, backup_time, "font.pak")
     # if not os.path.isdir(backup_dir):
       # os.makedirs(backup_dir)
     
     # Copy our existing data into the backup directory.
     # shutil.copy(game_bmp, backup_dir)
     # shutil.copy(game_font, backup_dir)
     
     # Then replace it with the one we generated.
     shutil.copy(font_bmp, os.path.join(font_dir, game_bmp))
     shutil.copy(font_font, os.path.join(font_dir, game_font))
   
   if for_editor:
     editor_png  = os.path.join(common.editor_config.gfx_dir, "font", "Font01.png"  if gen_font1 else "Font02.png")
     editor_font = os.path.join(common.editor_config.gfx_dir, "font", "Font01.font" if gen_font1 else "Font02.font")
     
     # Copy our files in.
     shutil.copy(font_png, editor_png)
     shutil.copy(font_font, editor_font)
     
     # Reparse the font for the editor.
     text_printer.load_fonts()
   
   shutil.rmtree(temp_dir)
コード例 #2
0
def import_umdimage(src, dst, convert_png=True, propogate=True, parent=None):
    src = os.path.abspath(src)
    dst = os.path.abspath(dst)
    if os.path.normcase(src) == os.path.normcase(dst):
        raise ValueError(
            "Cannot import %s. Source and destination directories are the same."
            % src)

    answer = QtGui.QMessageBox.question(
        parent,
        "Import Directory",
        "Importing directory:\n\n" + src + "\n\n" + "into directory:\n\n" +
        dst + "\n\n" + "Any affected files will be backed up. Proceed?",
        buttons=QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
        defaultButton=QtGui.QMessageBox.No)

    if answer == QtGui.QMessageBox.No:
        return

    progress = QProgressDialog("Finding files...", "Cancel", 0, 1, parent)
    progress.setWindowTitle("Importing...")
    progress.setWindowModality(Qt.Qt.WindowModal)
    progress.setValue(0)
    progress.setAutoClose(False)
    progress.setMinimumDuration(0)

    if parent:
        width = parent.width()
        height = parent.height()
        x = parent.x()
        y = parent.y()
    else:
        width = 1920
        height = 1080
        x = 0
        y = 0

    progress.setMaximum(0)
    progress.setValue(0)

    # The raw list of files we're importing.
    files = []

    # A list of lists, including all dupes of the files being imported, too.
    affected_files = []
    file_count = 0

    dupe_base = "umdimage"
    tmp = tempfile.mkdtemp(prefix="sdse-")

    seen_groups = []

    count = 0
    last_update = time.time()

    for file in list_all_files(src):
        if progress.wasCanceled():
            break

        # Strip our base directory, so we have just a relative file list.
        file = os.path.normpath(os.path.normcase(file[len(src) + 1:]))
        files.append(file)

        count += 1
        if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
            last_update = time.time()
            progress.setLabelText("Finding files...\n" + file)
            # progress.setValue(count)
            progress.setValue(progress.value() ^ 1)

            # Re-center the dialog.
            progress_w = progress.geometry().width()
            progress_h = progress.geometry().height()

            new_x = x + ((width - progress_w) / 2)
            new_y = y + ((height - progress_h) / 2)

            progress.move(new_x, new_y)

        affected_files.append([])

        if os.path.splitext(
                file)[1] == ".png" and convert_png and file not in SKIP_CONV:
            file = os.path.splitext(file)[0] + ".gim"

        if propogate:
            file_group = _DUPE_DB.group_from_file(os.path.join(
                dupe_base, file))
        else:
            file_group = None

        if file_group in seen_groups:
            continue

        # If there are no dupes, just add this file.
        if file_group == None:
            affected_files[-1].append(file)
            file_count += 1
            continue

        seen_groups.append(file_group)
        for dupe in _DUPE_DB.files_in_group(file_group):
            # Minus the "umdimage" part
            dupe = dupe[len(dupe_base) + 1:]
            affected_files[-1].append(dupe)
            file_count += 1

    progress.setValue(0)
    progress.setMaximum(file_count)

    # Make a backup first.
    backup_dir = None
    count = 0
    for file_set in affected_files:
        if progress.wasCanceled():
            break
        for file in file_set:
            if progress.wasCanceled():
                break
            count += 1
            if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
                last_update = time.time()
                progress.setLabelText("Backing up...\n" + file)
                progress.setValue(count)

                # Re-center the dialog.
                progress_w = progress.geometry().width()
                progress_h = progress.geometry().height()

                new_x = x + ((width - progress_w) / 2)
                new_y = y + ((height - progress_h) / 2)

                progress.move(new_x, new_y)

            # It's perfectly possible we want to import some files that
            # don't already exist. Such as when importing a directory
            # with added lines.
            if not os.path.isfile(os.path.join(dst, file)):
                continue

            backup_dir = backup_files(dst, [file],
                                      suffix="_IMPORT",
                                      backup_dir=backup_dir)

    progress.setValue(0)

    # And now do our importing.
    import_all_new = False
    skip_all_new = False
    count = 0
    for index, src_file in enumerate(files):
        if progress.wasCanceled():
            break

        if os.path.splitext(src_file)[
                1] == ".png" and convert_png and src_file not in SKIP_CONV:
            tmp_src_file = os.path.join(tmp, os.path.basename(src_file))
            tmp_src_file = os.path.splitext(tmp_src_file)[0] + ".gim"
            quantize = QuantizeType.auto
            for regex, q in FORCE_QUANTIZE:
                if not regex.search(src_file) == None:
                    quantize = q
                    break
            _CONV.png_to_gim(os.path.join(src, src_file), tmp_src_file,
                             quantize)
            src_file = tmp_src_file

        else:
            src_file = os.path.join(src, src_file)

        for file in affected_files[index]:
            if progress.wasCanceled():
                break

            dst_file = os.path.join(dst, file)

            count += 1
            # if count % 25 == 0:
            if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
                last_update = time.time()
                progress.setLabelText("Importing...\n" + file)
                progress.setValue(count)

                # Re-center the dialog.
                progress_w = progress.geometry().width()
                progress_h = progress.geometry().height()

                new_x = x + ((width - progress_w) / 2)
                new_y = y + ((height - progress_h) / 2)

                progress.move(new_x, new_y)

            # We may be allowed to import files that don't exist, but we're
            # going to ask them about it anyway.
            if not os.path.isfile(dst_file):
                if skip_all_new:
                    continue

                if not import_all_new:
                    answer = QtGui.QMessageBox.question(
                        parent,
                        "File Not Found",
                        "File:\n\n" + file + "\n\n" +
                        "does not exist in the target directory. Import anyway?",
                        buttons=QtGui.QMessageBox.Yes
                        | QtGui.QMessageBox.YesToAll | QtGui.QMessageBox.No
                        | QtGui.QMessageBox.NoToAll,
                        defaultButton=QtGui.QMessageBox.No)

                    if answer == QtGui.QMessageBox.YesToAll:
                        import_all_new = True
                        skip_all_new = False
                    elif answer == QtGui.QMessageBox.NoToAll:
                        skip_all_new = True
                        import_all_new = False
                        continue
                    elif answer == QtGui.QMessageBox.No:
                        continue

            basedir = os.path.dirname(dst_file)
            if not os.path.isdir(basedir):
                os.makedirs(basedir)

            shutil.copy2(src_file, dst_file)

    shutil.rmtree(tmp)
    progress.close()
コード例 #3
0
def import_umdimage2(src, dst, convert_png=True, propogate=True, parent=None):
    src = os.path.abspath(src)
    dst = os.path.abspath(dst)
    if os.path.normcase(src) == os.path.normcase(dst):
        raise ValueError(
            "Cannot import %s. Source and destination directories are the same."
            % src)

    answer = QtGui.QMessageBox.question(
        parent,
        "Import Directory",
        "Importing directory:\n\n" + src + "\n\n" + "into directory:\n\n" +
        dst + "\n\n" + "Any affected files will be backed up. Proceed?",
        buttons=QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
        defaultButton=QtGui.QMessageBox.No)

    if answer == QtGui.QMessageBox.No:
        return

    progress = QProgressDialog("Importing...", "Cancel", 0, 0, parent)
    progress.setWindowTitle("Importing...")
    progress.setWindowModality(Qt.Qt.WindowModal)
    progress.setValue(0)
    progress.setAutoClose(False)
    progress.setMinimumDuration(0)

    tmp_dst = tempfile.mkdtemp(prefix="sdse-")
    backup_dir = None

    for pak in glob.iglob(os.path.join(src, "bg_*.pak")):
        if progress.wasCanceled():
            break

        pak_name = os.path.basename(pak)
        backup_dir = backup_files(dst, [pak_name],
                                  suffix="_IMPORT",
                                  backup_dir=backup_dir)

        # If we have a regular file with the bg_*.pak name, then just drop it in.
        if os.path.isfile(pak):
            progress.setLabelText("Importing:\n" + pak_name)
            progress.setValue(progress.value() ^ 1)
            shutil.copy2(pak, os.path.join(dst, pak_name))

        # Otherwise, if it's a directory, insert all the textures we find
        # into the target bg_*.pak file.
        elif os.path.isdir(pak):
            for image in list_all_files(pak):
                if progress.wasCanceled():
                    break

                ext = os.path.splitext(image)[1].lower()
                if ext == ".png" and not convert_png:
                    continue

                base_name = image[len(src) + 1:]
                dst_files = []

                if propogate:
                    dupe_name = os.path.splitext(base_name)[0] + ".gim"
                    dupe_name = os.path.join("umdimage2", dupe_name)
                    dupe_name = os.path.normpath(os.path.normcase(dupe_name))

                    dupes = _DUPE_DB.files_in_same_group(dupe_name)

                    if dupes == None:
                        dupes = [dupe_name]

                    for dupe in dupes:
                        dst_file = dupe[10:]  # chop off the "umdimage2/"
                        dst_file = os.path.splitext(
                            dst_file)[0] + ext  # original extension
                        dst_file = os.path.join(tmp_dst, dst_file)
                        dst_files.append(dst_file)

                else:
                    dst_files = [os.path.join(tmp_dst, base_name)]

                for dst_file in dst_files:
                    try:
                        os.makedirs(os.path.dirname(dst_file))
                    except:
                        pass
                    shutil.copy(image, dst_file)

            if progress.wasCanceled():
                break

            progress.setLabelText("Inserting textures into:\n" + pak_name)
            progress.setValue(progress.value() ^ 1)

            pak_dir = os.path.join(tmp_dst, pak_name)
            pak_file = os.path.join(dst, pak_name)

            # If we didn't copy anything over, just move on.
            if not os.path.isdir(pak_dir):
                continue

            thread = threading.Thread(target=insert_textures,
                                      args=(pak_dir, pak_file))
            thread.start()

            while thread.isAlive():
                thread.join(MIN_INTERVAL)
                progress.setValue(progress.value() ^ 1)

                if progress.wasCanceled():
                    progress.setLabelText("Canceling...")

    shutil.rmtree(tmp_dst)
    progress.close()
コード例 #4
0
    def run(self):
        print "Really, no need for another Steam crawler"
        self.request_handler.start()
        health_check = True
        start_time = time.clock()
        queue_time = start_time
        dump_time = start_time - DATA_DUMP_TIME / 2 #first dump faster
        dump_time2 = start_time
        self.session_starttime = start_time
        self.dump_status(1)
        while not self.quit or not self.request_handler.done():
            if self.quit: self.request_handler.stop()

            if len(self.games_queue):
                game = self.games_queue.pop()
                print "Crawling game: " + game
                html = 1
                html1 = request_html("game " + game, "http://steamcommunity.com/app/" + game)
                if html1[0]: self.parse_game(html1[2], game)
                html2 = False,
            else:
                html = self.request_handler.get_html()
                if html != -1:
                    html1 = html[2]
                    html2 = html[3]
                    self.current_user = html[0]
                    self.current_url = "http://steamcommunity.com/" + self.current_user
                    if html1[0]: self.parse(html1[2], html2[2] if html2[0] else None, html[1])

                #stats
            current_time = time.time()
            if html != -1 and len(html1) > 1:
                self.save_times.append(current_time)
                html2_size = html2[1] if len(html2) > 1 else 0
                self.save_amounts.append(html1[1] + html2_size)
                self.alltimestats[1]+= 1
                self.alltimestats[2]+= html1[1] + html2_size

                #sleep
            end_time = time.clock()
            elapsed_time = end_time - start_time
            sleep_time = SLEEP_TIME - elapsed_time
            # print time until analysis
            time_until_analysis = DATA_DUMP_TIME - end_time + dump_time
            print "Analyzing in %i:%02i \r" % (time_until_analysis // 60, time_until_analysis % 60),
            # sleep now
            if sleep_time > 0:
                time.sleep(sleep_time)
                start_time = end_time + sleep_time
            else: start_time = end_time

                #performance stats
            if elapsed_time < ERROR_TIME and html != -1 and len(html1) > 1:
                self.crawl_times_sum+= elapsed_time
                self.crawl_times_amount+= 1

            if health_check:
                print "Crawling succesfully"
                health_check = False

                #data save/dump and backup
            if end_time - dump_time > DATA_DUMP_TIME: #sync
                self.request_handler.stop()
                if self.request_handler.done():
                    self.dump_data()
                    queue_time = time.clock()
                    dump_time = queue_time
                    dump_time2 = queue_time
                    self.request_handler.start()
            elif end_time - dump_time2 > STATUS_DUMP_TIME: #status
                self.dump_status()
                dump_time2 = end_time
            elif current_time > self.next_backup: #backup
                self.save_queue()
                self.next_backup = backup_files()
                queue_time = time.clock()
            elif end_time - queue_time > QUEUE_SAVE_TIME: #save
                self.save_queue()
                queue_time = time.clock()

        if self.quit_analyze: self.dump_data()
        else: self.dump_status()
        self.save_queue() #do this last!
コード例 #5
0
    def __generate_font__(self):
        gen_font1 = self.ui.rdoGenFont1.isChecked()
        temp_dir = tempfile.mkdtemp(prefix="sdse-")
        temp_name = "font"
        font_type = font_generator.FONT_TYPES.font01 if gen_font1 else font_generator.FONT_TYPES.font02
        for_game = self.ui.chkGenForGame.isChecked()
        for_editor = self.ui.chkGenForEditor.isChecked()

        font_data = self.get_font_data()
        if self.ui.rdoRightToLeft.isChecked():
            font_data.reverse()

        # Add the required characters to the end of the lowest-priority font
        # so they're there by default but they don't override any existing
        # settings for them, if they're already there.
        font_data[-1].chars += ''.join(REQUIRED_CHARS)

        font = font_generator.gen_font(font_data,
                                       font_type=font_type,
                                       img_width=1024,
                                       draw_outlines=False)
        font.save(temp_dir,
                  temp_name,
                  for_game,
                  for_editor,
                  font_type,
                  game=font_generator.GAMES.sdr2)

        basename = os.path.join(temp_dir, temp_name)
        font_png = basename + ".png"
        font_bmp = basename + ".bmp"
        font_font = basename + ".font"

        font_dir = os.path.join(common.editor_config.data01_dir, "jp", "font",
                                "font.pak")

        if for_game:
            game_bmp = "0000.bmp" if gen_font1 else "0002.bmp"
            game_font = "0001.font" if gen_font1 else "0003.font"

            backup_files(font_dir, [game_bmp, game_font], suffix="_FONT")
            # backup_time = time.strftime("%Y.%m.%d_%H.%M.%S_FONT")
            # backup_dir = os.path.join(common.editor_config.backup_dir, backup_time, "font.pak")
            # if not os.path.isdir(backup_dir):
            # os.makedirs(backup_dir)

            # Copy our existing data into the backup directory.
            # shutil.copy(game_bmp, backup_dir)
            # shutil.copy(game_font, backup_dir)

            # Then replace it with the one we generated.
            shutil.copy(font_bmp, os.path.join(font_dir, game_bmp))
            shutil.copy(font_font, os.path.join(font_dir, game_font))

        if for_editor:
            editor_png = os.path.join(
                common.editor_config.gfx_dir, "font",
                "Font01.png" if gen_font1 else "Font02.png")
            editor_font = os.path.join(
                common.editor_config.gfx_dir, "font",
                "Font01.font" if gen_font1 else "Font02.font")

            # Copy our files in.
            shutil.copy(font_png, editor_png)
            shutil.copy(font_font, editor_font)

            # Reparse the font for the editor.
            text_printer.load_fonts()

        shutil.rmtree(temp_dir)
コード例 #6
0
ファイル: crawl.py プロジェクト: Anaatti/Steam-crawler
    def run(self):
        print "Really, no need for another Steam crawler"
        self.request_handler.start()
        health_check = True
        start_time = time.clock()
        queue_time = start_time
        dump_time = start_time - DATA_DUMP_TIME / 2 #first dump faster
        dump_time2 = start_time
        self.session_starttime = start_time
        self.dump_status(1)
        while not self.quit or not self.request_handler.done():
            if self.quit: self.request_handler.stop()

            if len(self.games_queue):
                game = self.games_queue.pop()
                print "Crawling game: " + game
                html = 1
                html1 = request_html("game " + game, "http://steamcommunity.com/app/" + game)
                if html1[0]: self.parse_game(html1[2], game)
                html2 = False,
            else:
                html = self.request_handler.get_html()
                if html != -1:
                    html1 = html[2]
                    html2 = html[3]
                    self.current_user = html[0]
                    self.current_url = "http://steamcommunity.com/" + self.current_user
                    if html1[0]: self.parse(html1[2], html2[2] if html2[0] else None, html[1])

                #stats
            current_time = time.time()
            if html != -1 and len(html1) > 1:
                self.save_times.append(current_time)
                html2_size = html2[1] if len(html2) > 1 else 0
                self.save_amounts.append(html1[1] + html2_size)
                self.alltimestats[1]+= 1
                self.alltimestats[2]+= html1[1] + html2_size

                #sleep
            end_time = time.clock()
            elapsed_time = end_time - start_time
            sleep_time = SLEEP_TIME - elapsed_time
            # print time until analysis
            time_until_analysis = DATA_DUMP_TIME - end_time + dump_time
            print "Analyzing in %i:%02i \r" % (time_until_analysis // 60, time_until_analysis % 60),
            # sleep now
            if sleep_time > 0:
                time.sleep(sleep_time)
                start_time = end_time + sleep_time
            else: start_time = end_time

                #performance stats
            if elapsed_time < ERROR_TIME and html != -1 and len(html1) > 1:
                self.crawl_times_sum+= elapsed_time
                self.crawl_times_amount+= 1

            if health_check:
                print "Crawling succesfully"
                health_check = False

                #data save/dump and backup
            if end_time - dump_time > DATA_DUMP_TIME: #sync
                self.request_handler.stop()
                if self.request_handler.done():
                    self.dump_data()
                    queue_time = time.clock()
                    dump_time = queue_time
                    dump_time2 = queue_time
                    self.request_handler.start()
            elif end_time - dump_time2 > STATUS_DUMP_TIME: #status
                self.dump_status()
                dump_time2 = end_time
            elif current_time > self.next_backup: #backup
                self.save_queue()
                self.next_backup = backup_files()
                queue_time = time.clock()
            elif end_time - queue_time > QUEUE_SAVE_TIME: #save
                self.save_queue()
                queue_time = time.clock()

        if self.quit_analyze: self.dump_data()
        else: self.dump_status()
        self.save_queue() #do this last!
コード例 #7
0
def import_umdimage(src, dst, convert_png = True, propogate = True, parent = None):
  src = os.path.abspath(src)
  dst = os.path.abspath(dst)
  if os.path.normcase(src) == os.path.normcase(dst):
    raise ValueError("Cannot import %s. Source and destination directories are the same." % src)
    
  answer = QtGui.QMessageBox.question(
    parent,
    "Import Directory",
    "Importing directory:\n\n" + src + "\n\n" +
    "into directory:\n\n" + dst + "\n\n" +
    "Any affected files will be backed up. Proceed?",
    buttons = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
    defaultButton = QtGui.QMessageBox.No
  )
  
  if answer == QtGui.QMessageBox.No:
    return
  
  progress = QProgressDialog("Finding files...", "Cancel", 0, 1, parent)
  progress.setWindowTitle("Importing...")
  progress.setWindowModality(Qt.Qt.WindowModal)
  progress.setValue(0)
  progress.setAutoClose(False)
  progress.setMinimumDuration(0)
  
  if parent:
    width = parent.width()
    height = parent.height()
    x = parent.x()
    y = parent.y()
  else:
    width   = 1920
    height  = 1080
    x       = 0
    y       = 0
  
  progress.setMaximum(0)
  progress.setValue(0)
  
  # The raw list of files we're importing.
  files = []
  
  # A list of lists, including all dupes of the files being imported, too.
  affected_files = []
  file_count = 0
  
  dupe_base = "umdimage"
  tmp       = tempfile.mkdtemp(prefix = "sdse-")
  
  seen_groups = []
  
  count = 0
  last_update = time.time()
  
  for file in list_all_files(src):
    if progress.wasCanceled():
      break
    
    # Strip our base directory, so we have just a relative file list.
    file = os.path.normpath(os.path.normcase(file[len(src) + 1:]))
    files.append(file)
    
    count += 1
    if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
      last_update = time.time()
      progress.setLabelText("Finding files...\n" + file)
      # progress.setValue(count)
      progress.setValue(progress.value() ^ 1)
      
      # Re-center the dialog.
      progress_w = progress.geometry().width()
      progress_h = progress.geometry().height()
      
      new_x = x + ((width - progress_w) / 2)
      new_y = y + ((height - progress_h) / 2)
      
      progress.move(new_x, new_y)
    
    affected_files.append([])
    
    if os.path.splitext(file)[1] == ".png" and convert_png and file not in SKIP_CONV:
      file = os.path.splitext(file)[0] + ".gim"
    
    if propogate:
      file_group = _DUPE_DB.group_from_file(os.path.join(dupe_base, file))
    else:
      file_group = None
    
    if file_group in seen_groups:
      continue
    
    # If there are no dupes, just add this file.
    if file_group == None:
      affected_files[-1].append(file)
      file_count += 1
      continue
    
    seen_groups.append(file_group)
    for dupe in _DUPE_DB.files_in_group(file_group):
      # Minus the "umdimage" part
      dupe = dupe[len(dupe_base) + 1:]
      affected_files[-1].append(dupe)
      file_count += 1
  
  progress.setValue(0)
  progress.setMaximum(file_count)
  
  # Make a backup first.
  backup_dir = None
  count = 0
  for file_set in affected_files:
    if progress.wasCanceled():
      break
    for file in file_set:
      if progress.wasCanceled():
        break
      count += 1
      if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
        last_update = time.time()
        progress.setLabelText("Backing up...\n" + file)
        progress.setValue(count)
        
        # Re-center the dialog.
        progress_w = progress.geometry().width()
        progress_h = progress.geometry().height()
        
        new_x = x + ((width - progress_w) / 2)
        new_y = y + ((height - progress_h) / 2)
        
        progress.move(new_x, new_y)
      
      # It's perfectly possible we want to import some files that
      # don't already exist. Such as when importing a directory
      # with added lines.
      if not os.path.isfile(os.path.join(dst, file)):
        continue
        
      backup_dir = backup_files(dst, [file], suffix = "_IMPORT", backup_dir = backup_dir)
  
  progress.setValue(0)
  
  # And now do our importing.
  import_all_new = False
  skip_all_new = False
  count = 0
  for index, src_file in enumerate(files):
    if progress.wasCanceled():
      break
    
    if os.path.splitext(src_file)[1] == ".png" and convert_png and src_file not in SKIP_CONV:
      tmp_src_file = os.path.join(tmp, os.path.basename(src_file))
      tmp_src_file = os.path.splitext(tmp_src_file)[0] + ".gim"
      quantize = QuantizeType.auto
      for regex, q in FORCE_QUANTIZE:
        if not regex.search(src_file) == None:
          quantize = q
          break
      _CONV.png_to_gim(os.path.join(src, src_file), tmp_src_file, quantize)
      src_file = tmp_src_file
    
    else:
      src_file = os.path.join(src, src_file)
    
    for file in affected_files[index]:
      if progress.wasCanceled():
        break
      
      dst_file = os.path.join(dst, file)
      
      count += 1
      # if count % 25 == 0:
      if time.time() - last_update > MIN_INTERVAL or count % 25 == 0:
        last_update = time.time()
        progress.setLabelText("Importing...\n" + file)
        progress.setValue(count)
        
        # Re-center the dialog.
        progress_w = progress.geometry().width()
        progress_h = progress.geometry().height()
        
        new_x = x + ((width - progress_w) / 2)
        new_y = y + ((height - progress_h) / 2)
        
        progress.move(new_x, new_y)
      
      # We may be allowed to import files that don't exist, but we're
      # going to ask them about it anyway.
      if not os.path.isfile(dst_file):
        if skip_all_new:
          continue
        
        if not import_all_new:
          answer = QtGui.QMessageBox.question(
            parent,
            "File Not Found",
            "File:\n\n" + file + "\n\n" +
            "does not exist in the target directory. Import anyway?",
            buttons = QtGui.QMessageBox.Yes | QtGui.QMessageBox.YesToAll | QtGui.QMessageBox.No | QtGui.QMessageBox.NoToAll,
            defaultButton = QtGui.QMessageBox.No
          )
          
          if answer == QtGui.QMessageBox.YesToAll:
            import_all_new = True
            skip_all_new = False
          elif answer == QtGui.QMessageBox.NoToAll:
            skip_all_new = True
            import_all_new = False
            continue
          elif answer == QtGui.QMessageBox.No:
            continue
      
      basedir = os.path.dirname(dst_file)
      if not os.path.isdir(basedir):
        os.makedirs(basedir)
      
      shutil.copy2(src_file, dst_file)
  
  shutil.rmtree(tmp)
  progress.close()
コード例 #8
0
def import_umdimage2(src, dst, convert_png = True, propogate = True, parent = None):
  src = os.path.abspath(src)
  dst = os.path.abspath(dst)
  if os.path.normcase(src) == os.path.normcase(dst):
    raise ValueError("Cannot import %s. Source and destination directories are the same." % src)
    
  answer = QtGui.QMessageBox.question(
    parent,
    "Import Directory",
    "Importing directory:\n\n" + src + "\n\n" +
    "into directory:\n\n" + dst + "\n\n" +
    "Any affected files will be backed up. Proceed?",
    buttons = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
    defaultButton = QtGui.QMessageBox.No
  )
  
  if answer == QtGui.QMessageBox.No:
    return
  
  progress = QProgressDialog("Importing...", "Cancel", 0, 0, parent)
  progress.setWindowTitle("Importing...")
  progress.setWindowModality(Qt.Qt.WindowModal)
  progress.setValue(0)
  progress.setAutoClose(False)
  progress.setMinimumDuration(0)
  
  tmp_dst     = tempfile.mkdtemp(prefix = "sdse-")
  backup_dir  = None
  
  for pak in glob.iglob(os.path.join(src, "bg_*.pak")):
    if progress.wasCanceled():
      break
    
    pak_name    = os.path.basename(pak)
    backup_dir  = backup_files(dst, [pak_name], suffix = "_IMPORT", backup_dir = backup_dir)
    
    # If we have a regular file with the bg_*.pak name, then just drop it in.
    if os.path.isfile(pak):
      progress.setLabelText("Importing:\n" + pak_name)
      progress.setValue(progress.value() ^ 1)
      shutil.copy2(pak, os.path.join(dst, pak_name))
    
    # Otherwise, if it's a directory, insert all the textures we find
    # into the target bg_*.pak file.
    elif os.path.isdir(pak):
      for image in list_all_files(pak):
        if progress.wasCanceled():
          break
        
        ext = os.path.splitext(image)[1].lower()
        if ext == ".png" and not convert_png:
          continue
        
        base_name = image[len(src) + 1:]
        dst_files = []
        
        if propogate:
          dupe_name = os.path.splitext(base_name)[0] + ".gim"
          dupe_name = os.path.join("umdimage2", dupe_name)
          dupe_name = os.path.normpath(os.path.normcase(dupe_name))
        
          dupes = _DUPE_DB.files_in_same_group(dupe_name)
          
          if dupes == None:
            dupes = [dupe_name]
          
          for dupe in dupes:
            dst_file = dupe[10:] # chop off the "umdimage2/"
            dst_file = os.path.splitext(dst_file)[0] + ext # original extension
            dst_file = os.path.join(tmp_dst, dst_file)
            dst_files.append(dst_file)
        
        else:
          dst_files = [os.path.join(tmp_dst, base_name)]
        
        for dst_file in dst_files:
          try:
            os.makedirs(os.path.dirname(dst_file))
          except:
            pass
          shutil.copy(image, dst_file)
      
      if progress.wasCanceled():
        break
    
      progress.setLabelText("Inserting textures into:\n" + pak_name)
      progress.setValue(progress.value() ^ 1)
      
      pak_dir   = os.path.join(tmp_dst, pak_name)
      pak_file  = os.path.join(dst, pak_name)
      
      # If we didn't copy anything over, just move on.
      if not os.path.isdir(pak_dir):
        continue
      
      thread = threading.Thread(target = insert_textures, args = (pak_dir, pak_file))
      thread.start()
      
      while thread.isAlive():
        thread.join(MIN_INTERVAL)
        progress.setValue(progress.value() ^ 1)
        
        if progress.wasCanceled():
          progress.setLabelText("Canceling...")
  
  shutil.rmtree(tmp_dst)
  progress.close()