58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
import os
|
|
from datetime import datetime
|
|
from app import create_app
|
|
from models import db, Room, RoomFile, User
|
|
|
|
DATA_ROOT = '/data/rooms'
|
|
|
|
app = create_app()
|
|
|
|
with app.app_context():
|
|
admin_user = User.query.filter_by(is_admin=True).first()
|
|
if not admin_user:
|
|
print('No admin user found. Aborting.')
|
|
exit(1)
|
|
rooms = Room.query.all()
|
|
for room in rooms:
|
|
room_dir = os.path.join(DATA_ROOT, str(room.id))
|
|
if not os.path.exists(room_dir):
|
|
continue
|
|
for root, dirs, files in os.walk(room_dir):
|
|
rel_root = os.path.relpath(root, room_dir)
|
|
rel_path = '' if rel_root == '.' else rel_root.replace('\\', '/')
|
|
# Folders
|
|
for d in dirs:
|
|
exists = RoomFile.query.filter_by(room_id=room.id, name=d, path=rel_path, type='folder').first()
|
|
folder_path = os.path.join(root, d)
|
|
stat = os.stat(folder_path)
|
|
if not exists:
|
|
rf = RoomFile(
|
|
room_id=room.id,
|
|
name=d,
|
|
path=rel_path,
|
|
type='folder',
|
|
size=None,
|
|
modified=stat.st_mtime,
|
|
uploaded_by=admin_user.id,
|
|
uploaded_at=datetime.utcfromtimestamp(stat.st_mtime)
|
|
)
|
|
db.session.add(rf)
|
|
# Files
|
|
for f in files:
|
|
exists = RoomFile.query.filter_by(room_id=room.id, name=f, path=rel_path, type='file').first()
|
|
file_path = os.path.join(root, f)
|
|
stat = os.stat(file_path)
|
|
if not exists:
|
|
rf = RoomFile(
|
|
room_id=room.id,
|
|
name=f,
|
|
path=rel_path,
|
|
type='file',
|
|
size=stat.st_size,
|
|
modified=stat.st_mtime,
|
|
uploaded_by=admin_user.id,
|
|
uploaded_at=datetime.utcfromtimestamp(stat.st_mtime)
|
|
)
|
|
db.session.add(rf)
|
|
db.session.commit()
|
|
print('RoomFile table synchronized with filesystem.') |