def get_files(index):
a = select(
'SELECT * FROM files ' + 'ORDER BY file_id DESC LIMIT 5 OFFSET ?',
index)
for i in a:
i['name'] = get_album_name(i['album_id'])
return a
def get_user_files(user_id):
a = select(
'SELECT * FROM files WHERE user_id=? ORDER BY file_id DESC LIMIT 50',
user_id)
for i in a:
i['name'] = get_album_name(i['album_id'])
return a
def get_file(file_id, ptype):
a = select('SELECT * FROM files WHERE file_id=?', file_id)
if len(a) == 0:
return None
p = a[0]
p['path'] = photo_file.get_path(file_id)
if ptype == 'thumb':
p['path'] = photo_file.make_thubnail(file_id, 300)
return p
def get_file(file_id, ptype):
# データベースから基本情報を得る
a = select('SELECT * FROM files WHERE file_id=?', file_id)
if len(a) == 0: return None
p = a[0]
p['path'] = photo_file.get_path(file_id)
# サムネイル画像の指定であれば作成する --- (*10)
if ptype == 'thumb':
p['path'] = photo_file.make_thumbnail(file_id, 300)
return p
def save_file(user_id, upfile, album_id):
if not re.search(r'\.(jpg|jpeg)$', upfile.filename):
print('JPEGではない：', upfile.filename)
return 0
if album_id == 0:
a = select('SELECT * FROM albums WHERE user_id=? AND name=?', user_id,
'未分類')
if len(a) == 0:
album_id = exec('INSERT INTO albums (user_id, name) VALUES (?,?)',
user_id, '未分類')
else:
album_id = a[0]['album_id']
file_id = exec(
'''
INSERT INTO files (user_id, filename, album_id) VALUES (?,?,?)
''', user_id, upfile.filename, album_id)
upfile.save(photo_file.get_path(file_id))
return file_id
def save_file(user_id, upfile, album_id):
# JPEGファイルだけを許可
if not re.search(r'\.(jpg|jpeg)$', upfile.filename):
print('JPEGではない:', upfile.filename)
return 0
# アルバム未指定の場合、未分類アルバムを自動的に作る
if album_id == 0:
a = select('SELECT * FROM albums WHERE user_id=? AND name=?', user_id,
'未分類')
if len(a) == 0:
album_id = exec('INSERT INTO albums (user_id, name) VALUES(?, ?)',
user_id, '未分類')
else:
album_id = a[0]['album_id']
# ファイル情報を保存
file_id = exec(
'''
INSERT INTO files(user_id, filename, album_id)
VALUES(?, ?, ?)''', user_id, upfile.filename, album_id)
# ファイルを保存
upfile.save(photo_file.get_path(file_id))
return file_id
def get_friend_list():
return select('SELECT user_id,friend FROM users ')
def get_user(user_id):
return select('SELECT * FROM users WHERE user_id=?', user_id)
def get_album(album_id):
a = select('SELECT * FROM albums WHERE album_id=?', album_id)
if len(a) == 0: return None
return a[0]
def get_albums(user_id):
return select('SELECT * FROM albums WHERE user_id=?', user_id)
def get_album_files(album_id):
return select(
'''
SELECT * FROM files WHERE album_id=?
ORDER BY file_id DESC''', album_id)
def get_amount():
a = select('SELECT count(*) FROM files')
b = a[0]['count(*)']
return b
def get_files():
a = select('SELECT * FROM files ORDER BY file_id DESC LIMIT 50')
for i in a:
i['name'] = get_album_name(i['album_id'])
return a
