35 lines
788 B
Python
35 lines
788 B
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Makes all existing filepaths in the database relative to `config.music_dir`.
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
import asyncpg
|
||
|
|
|
||
|
|
import config
|
||
|
|
|
||
|
|
|
||
|
|
async def fix_fpaths():
|
||
|
|
"""
|
||
|
|
Makes all existing filepaths in the database relative to
|
||
|
|
`config.music_dir`.
|
||
|
|
"""
|
||
|
|
db_pool = await asyncpg.create_pool(**config.db)
|
||
|
|
async with db_pool.acquire() as conn:
|
||
|
|
tracks = await conn.fetch("SELECT filepath FROM track")
|
||
|
|
|
||
|
|
data = []
|
||
|
|
for track in tracks:
|
||
|
|
fpath = track['filepath']
|
||
|
|
fpath = os.path.relpath(fpath, config.music_dir)
|
||
|
|
entry = (fpath, track['filepath'])
|
||
|
|
data.append(entry)
|
||
|
|
|
||
|
|
stm = await conn.prepare("UPDATE track SET filepath = $1 WHERE filepath = $2")
|
||
|
|
await stm.executemany(data)
|
||
|
|
print("Done.")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
asyncio.run(fix_fpaths())
|