forked from LBRYCommunity/lbry-sdk
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import os
|
|
import binascii
|
|
import sqlite3
|
|
from lbrynet.conf import Config
|
|
|
|
|
|
def main():
|
|
conf = Config()
|
|
db = sqlite3.connect(os.path.join(conf.data_dir, 'lbrynet.sqlite'))
|
|
cur = db.cursor()
|
|
files = cur.execute("select stream_hash, file_name, download_directory from file").fetchall()
|
|
update = {}
|
|
for stream_hash, file_name, download_directory in files:
|
|
try:
|
|
binascii.unhexlify(file_name)
|
|
except binascii.Error:
|
|
try:
|
|
binascii.unhexlify(download_directory)
|
|
except binascii.Error:
|
|
update[stream_hash] = (
|
|
binascii.hexlify(file_name.encode()).decode(), binascii.hexlify(download_directory.encode()).decode()
|
|
)
|
|
if update:
|
|
print(f"repair {len(update)} streams")
|
|
for stream_hash, (file_name, download_directory) in update.items():
|
|
cur.execute('update file set file_name=?, download_directory=? where stream_hash=?',
|
|
(file_name, download_directory, stream_hash))
|
|
db.commit()
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|