Browse Source

Newline between functios

restore_refactor
Raphael Roberts 7 years ago
parent
commit
37f2f17ff3
  1. 3
      backup.py
  2. 8
      restore.py

3
backup.py

@ -53,18 +53,21 @@ def hashify(top):
print(e,file) print(e,file)
os.chdir(old_dir) os.chdir(old_dir)
return ret return ret
def __init_database__(path): def __init_database__(path):
con = sqlite3.connect(path) con = sqlite3.connect(path)
cur = con.cursor() cur = con.cursor()
cur.execute('CREATE TABLE IF NOT EXISTS `paths` (`path` TEXT, `imohash` BLOB, `blake` BLOB, UNIQUE(`path`, `imohash`, `blake`));') cur.execute('CREATE TABLE IF NOT EXISTS `paths` (`path` TEXT, `imohash` BLOB, `blake` BLOB, UNIQUE(`path`, `imohash`, `blake`));')
con.commit() con.commit()
return con return con
def backup(top,db_path): def backup(top,db_path):
paths = hashify(top) paths = hashify(top)
con = __init_database__(db_path) con = __init_database__(db_path)
cur = con.cursor() cur = con.cursor()
cur.executemany('INSERT OR IGNORE INTO `paths` VALUES (?,?,?);',paths) cur.executemany('INSERT OR IGNORE INTO `paths` VALUES (?,?,?);',paths)
con.commit() con.commit()
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('dir') parser.add_argument('dir')

8
restore.py

@ -11,14 +11,15 @@ def create_temp():
name= 'fsb{}'.format(hex(time.time_ns())[2:]) name= 'fsb{}'.format(hex(time.time_ns())[2:])
os.mkdir(name) os.mkdir(name)
return name return name
def lookup(cur,imohash,realhash=None): def lookup(cur,imohash,realhash=None):
if realhash is not None: if realhash is not None:
cur.execute('SELECT path FROM paths where blake=?',[realhash]) cur.execute('SELECT path FROM paths where blake=?',[realhash])
else: else:
cur.execute('SELECT path FROM paths where imohash=?',[imohash]) cur.execute('SELECT path FROM paths where imohash=?',[imohash])
res = cur.fetchone()
if len(res) == 1:
return res[0]
res = cur.fetchall()
return res
def restore(database,source,destination): def restore(database,source,destination):
con=sqlite3.connect(database) con=sqlite3.connect(database)
cur=con.cursor() cur=con.cursor()
@ -34,6 +35,7 @@ def restore(database,source,destination):
print(ppath,qpath,sep = '->') print(ppath,qpath,sep = '->')
os.rename(ppath,qpath) os.rename(ppath,qpath)
os.rename(temp,destination) os.rename(temp,destination)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('database') parser.add_argument('database')

Loading…
Cancel
Save