def test_if_domain_id_exists(cls, domain_id):
sql = "select count(domain_id) from domain where domain_id=%s"
rows = db_conn.query_all(sql, domain_id)
if rows[0][0] == 1:
return True
else:
return False
def get_domain_by_domain_name(cls, domain_name):
sql = "select domain_id, domain_name, type from domain where domain_name=%s"
rows = db_conn.query_all(sql, domain_name)
if len(rows) == 1:
return rows[0]
else:
return False
def get_userinfo_by_user_id(cls, user_id):
sql = "select user_id, username, md5_password from userinfo where user_id=%s"
rows = db_conn.query_all(sql, user_id)
if len(rows) == 1:
return rows[0]
else:
return False
def get_user_domain_by_user_id_and_domain_id(cls, user_id, domain_id):
sql = "select id, user_id, domain_id from user_domain where user_id=%s and domain_id=%s"
rows = db_conn.query_all(sql, (user_id, domain_id))
return rows
def get_all_domain(cls):
sql = "select domain_id, domain_name, type from domain order by domain_name"
rows = db_conn.query_all(sql)
return rows
def get_dns_record_list_by_domain_id(cls, domain_id):
sql = "select id, domain_id, record, ttl, type, value, create_time, state from dns_record where domain_id=%s"
rows = db_conn.query_all(sql, domain_id)
return rows
def type_cname_count(cls, domain_id, record, value):
sql = "select count(*) from dns_record where domain_id=%s and record=%s and value=%s and type='CNAME'"
rows = db_conn.query_all(sql, (domain_id, record, value))
return rows[0][0]
def type_a_all_count(cls, domain_id, record):
sql = "select count(*) from dns_record where domain_id=%s and record=%s and type='A'"
rows = db_conn.query_all(sql, (domain_id, record))
return rows[0][0]
def change_dns_record_state(cls, domain_id, record_id, state):
sql = "update dns_record set state=%s where domain_id=%s and id=%s"
rows = db_conn.query_all(sql, (state, domain_id, record_id))
return True
def del_dns_record(cls, domain_id, record_id):
sql = "delete from dns_record where domain_id=%s and id=%s"
rows = db_conn.query_all(sql, (domain_id, record_id))
return True
def add_dns_record(cls, domain_id, record, ttl, type, value, line):
sql = "insert into dns_record (domain_id, record, ttl, type, value, create_time, state, line) values (%s, %s, %s, %s, %s, now(), '1', %s)"
rows = db_conn.insert(sql, (domain_id, record, ttl, type, value, line))
sql = "select max(id) from dns_record"
rows = db_conn.query_all(sql)
return rows[0][0]