def album_new(user_id, args):
name = args.get('name', '')
if name == '':
return 0
album_id = exec('INSERT INTO albums (name, user_id) VALUES (?,?)', name,
user_id)
return album_id
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
# テーブルを作成するSQLの実行
from photo_sqlite import exec
exec('''
CREATE TABLE files (
file_id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id TEXT,
filename TEXT,
album_id INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT(DATETIME('now', 'localtime'))
)
''')
exec('''
CREATE TABLE albums (
album_id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT,
user_id TEXT,
created_at TIMESTAMP DEFAULT(DATETIME('now', 'localtime'))
)
''')
print('ok')
def public_c(file_id):
exec('UPDATE files SET public= ~public WHERE file_id=?', file_id)
def change_album(album_id, file_id):
exec('UPDATE files SET album_id=? WHERE file_id=?', album_id, file_id)
def delete_photo(file_id):
exec('DELETE FROM files WHERE file_id=?', file_id)
def friend_delete(user_id, args):
exec('UPDATE users SET friend=? WHERE user_id=?', args, user_id)
def friend_add(user_id, args):
exec('UPDATE users SET friend = (friend||?) WHERE user_id = ?', args,
user_id)
def delete_file(file_id):
a = exec('''
DELETE FROM files WHERE file_id=?
''', file_id)
return a
from photo_sqlite import exec
#ユーザー生成の時にidを付ける
exec('''
/* ユーザー情報 */
CREATE TABLE users (
login_id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id TEXT,
friend TEXT DEFAULT('')
)
''')
exec('''
/* ユーザー情報 */
INSERT INTO users
(user_id)
VALUES
("taro"),("jiro"),("sabu"),("siro"),("goro");
''')
exec('''
/* ファイル情報 */
CREATE TABLE files (
file_id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id TEXT,
filename TEXT,
