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)
os.chdir(old_dir)
return ret
def __init_database__(path):
con = sqlite3.connect(path)
cur = con.cursor()
cur.execute('CREATE TABLE IF NOT EXISTS `paths` (`path` TEXT, `imohash` BLOB, `blake` BLOB, UNIQUE(`path`, `imohash`, `blake`));')
con.commit()
return con
def backup(top,db_path):
paths = hashify(top)
con = __init_database__(db_path)
cur = con.cursor()
cur.executemany('INSERT OR IGNORE INTO `paths` VALUES (?,?,?);',paths)
con.commit()
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('dir')

8
restore.py

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

Loading…
Cancel
Save