#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys, os
import zipfile
import datetime
import xml.dom.minidom
# 7 avg 2021
# Запись списка книг в "небазу" (третья,2 версия)
#--------------------------------------------------
Books = []
Errror = []
siz = 0 # размер файла
countFiles = 0
book_title = ''
stroka = ''
replace_val = [('\\', '_'),('/', '_'),(':', '_'),('*', '_'),('?', '_'),('"', '_'),('<', '_'), ('>', '_'), ('|', '_')]
#--------------------------------------------------
def SaveList(fn, L, cco):
, ,if cco > 0:
, , , ,f = open(fn, 'w')
, , , ,for i in L:
, , , , , ,f.write(i)
, , , ,f.close()
def SaveError(fn, L):
, ,if len(L) > 0:
, , , ,f = open(fn, 'w')
, , , ,for i in L:
, , , , , ,f.write(str(i)+'\n')
, , , ,f.close()
#--------------------------------------------------
#**********************************************************************
def parse_fb2(fn):
, ,global Books
, ,global stroka
, ,global book_title
, , , ,
, ,if isinstance(fn, str):
, , , ,fn = open(fn)
, ,try:
, , , ,dom = xml.dom.minidom.parse(fn)
, , , ,
, ,except:
, , , ,print('Error in fb2:')
, , , ,Errror.append(fn)
, , , ,return False
, ,else:
, , , ,dom.normalize()
, , , ,node1=dom.getElementsByTagName("description")[0]
, , , ,titl=node1.getElementsByTagName("title-info")[0]
##
, , , ,book=titl.getElementsByTagName("book-title")[0]
, , , ,book_title = book.childNodes[0].nodeValue
, , , ,book_title = book_title.replace('|', '!')
, , , ,s = book_title +'|'
, , , ,
, , , ,au = ''
, , , ,try:
, , , , , ,for auto in titl.getElementsByTagName("author"):
, , , , , , , ,no1=auto.getElementsByTagName("last-name")[0]
, , , , , , , ,au = au + no1.childNodes[0].nodeValue
, , , , , , , ,no1=auto.getElementsByTagName("first-name")[0]
, , , , , , , ,fi = no1.childNodes[0].nodeValue
, , , , , , , ,au = au + '.'+ fi[0]+'^' , , , , , ,
, , , , , ,au = au[:-1]
, , , ,except:
, , , , , ,au = "NoName."
, , , ,s = s+au
, , , ,stroka = s +'|0|'
, , , ,return True
#**********************************************************************
#--------------------------------------------
def parse_zip(adr):
, ,z = zipfile.ZipFile(adr, 'r')
, ,filelist = z.namelist()
, ,filelist.sort()
, ,for n in filelist:
, , , , try:
, , , , , ,if n[-4:] == ".fb2":
, , , , , , , ,return parse_fb2(z.open(n))
, , , , except:
, , , , , ,print( "Errror:", n )
, , , , , ,return False
#--------------------------------------------------
def replace(line, old_new_num):
, ,for vals in old_new_num:
, , , , , ,# распаковываем кортеж
, , , , , ,old, new = vals
, , , , , ,line = line.replace(old, new)
, ,return line
def my_rename(adr,ph, ex):
, ,co = 0
, ,book_ti = book_title[:20]
# \ / : * ? " < > |
, ,book_ti = replace(book_ti, replace_val)
, ,
, ,a = '-' + book_ti + ex
, ,while os.path.isfile(os.path.join(ph, a)):
, , , ,co += 1
, , , ,a = '-' + book_ti +str(co)+ ex
# , ,print(a)
, ,os.rename(adr,os.path.join(ph, a))
, ,if ex == '.fb2':
, , , ,ex = 'f'
, ,else:
, , , ,ex = 'z'
, ,if co > 0:
, , , ,ex = str(co)+ex
, ,
, ,Books.append(stroka +ex+ '|'+str(siz)+'\n')
, ,
def parse_file(adr,ph):
, ,global siz # размер файла
, ,global countFiles
, ,flag = os.path.basename(adr)[0] != '-'
, ,siz = os.path.getsize(adr)
, ,m = adr.split(".")
, ,if (m[-1] == "zip") and (m[-2] == "fb2"):
, , , ,if flag and parse_zip(adr):
, , , , , , countFiles += 1
, , , , , , my_rename(adr,ph,'.fb2.zip')
, ,elif (m[-1] == "fb2"):
, , , ,if flag and parse_fb2(adr):
, , , , , , countFiles += 1
, , , , , , my_rename(adr,ph,'.fb2') , , , , , ,
, , , ,
def parse_dir(ph):
, ,dirlist = os.listdir(ph)
, ,dirlist.sort()
, ,for i in dirlist:
, , , ,adr = os.path.join(ph, i)
, , , ,if os.path.isdir(adr):
, , , , , ,parse_dir(adr)
, , , ,else:
#? , , , , , ,if os.path.getsize(i) > 0:
, , , , , , , ,parse_file(adr, ph)
#? , , , ,else:
#? , , , , , ,print("bad zip "+ i)
#----------------------------------------
path = os.getcwd()
, ,
#----------------------------------------
if os.path.isfile('NeBaza.txt'):
, ,f = open('NeBaza.txt','r')
, ,for i in f:
, , , , , ,Books.append(i)
, ,f.close()
parse_dir(path)
Books.sort()
SaveList('NeBaza.txt', Books, countFiles )
SaveError('Errror.txt', Errror)#
print( 'Done!')
print( 'Add '+ str(countFiles)+' files.')