| import logging |
import logging |
| |
|
| import ckan.plugins as plugins |
import ckan.plugins as plugins |
| import ckan.lib as lib |
import ckan.lib as lib |
| import ckan.lib.dictization.model_dictize as model_dictize |
import ckan.lib.dictization.model_dictize as model_dictize |
| import ckan.plugins.toolkit as tk |
import ckan.plugins.toolkit as tk |
| import ckan.model as model |
import ckan.model as model |
| from pylons import config |
from pylons import config |
| |
|
| from sqlalchemy import orm |
from sqlalchemy import orm |
| import ckan.model |
import ckan.model |
| |
|
| # get user created datasets and those they have edited |
# get user created datasets and those they have edited |
| def get_user_datasets(user_dict): |
def get_user_datasets(user_dict): |
| created_datasets_list = user_dict['datasets'] |
created_datasets_list = user_dict['datasets'] |
| active_datasets_list = [x['data']['package'] for x in |
active_datasets_list = [x['data']['package'] for x in |
| lib.helpers.get_action('user_activity_list',{'id':user_dict['id']}) if x['data'].get('package')] |
lib.helpers.get_action('user_activity_list',{'id':user_dict['id']}) if x['data'].get('package')] |
| raw_list = created_datasets_list + active_datasets_list |
raw_list = created_datasets_list + active_datasets_list |
| filtered_dict = {} |
filtered_dict = {} |
| for dataset in raw_list: |
for dataset in raw_list: |
| if dataset['id'] not in filtered_dict.keys(): |
if dataset['id'] not in filtered_dict.keys(): |
| filtered_dict[dataset['id']] = dataset |
filtered_dict[dataset['id']] = dataset |
| return filtered_dict.values() |
return filtered_dict.values() |
| |
|
| def get_related_dataset(related_id): |
def get_related_dataset(related_id): |
| result = model.Session.execute("select dataset_id from related_dataset where related_id =\'"+related_id+"\' limit 1;").first()[0] |
result = model.Session.execute("select dataset_id from related_dataset where related_id =\'"+related_id+"\' limit 1;").first()[0] |
| return lib.helpers.get_action('package_show',{'id':result}) |
return result |
| |
|
| def related_create(context, data_dict=None): |
def related_create(context, data_dict=None): |
| return {'success': False, 'msg': 'No one is allowed to create related items'} |
return {'success': False, 'msg': 'No one is allowed to create related items'} |
| |
|
| class DataGovAuPlugin(plugins.SingletonPlugin, |
class DataGovAuPlugin(plugins.SingletonPlugin, |
| tk.DefaultDatasetForm): |
tk.DefaultDatasetForm): |
| '''An example IDatasetForm CKAN plugin. |
'''An example IDatasetForm CKAN plugin. |
| |
|
| Uses a tag vocabulary to add a custom metadata field to datasets. |
Uses a tag vocabulary to add a custom metadata field to datasets. |
| |
|
| ''' |
''' |
| plugins.implements(plugins.IConfigurer, inherit=False) |
plugins.implements(plugins.IConfigurer, inherit=False) |
| plugins.implements(plugins.ITemplateHelpers, inherit=False) |
plugins.implements(plugins.ITemplateHelpers, inherit=False) |
| plugins.implements(plugins.IAuthFunctions) |
plugins.implements(plugins.IAuthFunctions) |
| |
|
| def get_auth_functions(self): |
def get_auth_functions(self): |
| return {'related_create': related_create} |
return {'related_create': related_create} |
| |
|
| def update_config(self, config): |
def update_config(self, config): |
| # Add this plugin's templates dir to CKAN's extra_template_paths, so |
# Add this plugin's templates dir to CKAN's extra_template_paths, so |
| # that CKAN will use this plugin's custom templates. |
# that CKAN will use this plugin's custom templates. |
| # here = os.path.dirname(__file__) |
# here = os.path.dirname(__file__) |
| # rootdir = os.path.dirname(os.path.dirname(here)) |
# rootdir = os.path.dirname(os.path.dirname(here)) |
| |
|
| tk.add_template_directory(config, 'templates') |
tk.add_template_directory(config, 'templates') |
| tk.add_public_directory(config, 'theme/public') |
tk.add_public_directory(config, 'theme/public') |
| tk.add_resource('theme/public', 'ckanext-datagovau') |
tk.add_resource('theme/public', 'ckanext-datagovau') |
| # config['licenses_group_url'] = 'http://%(ckan.site_url)/licenses.json' |
# config['licenses_group_url'] = 'http://%(ckan.site_url)/licenses.json' |
| |
|
| def get_helpers(self): |
def get_helpers(self): |
| return {'get_user_datasets': get_user_datasets, 'get_related_dataset': get_related_dataset} |
return {'get_user_datasets': get_user_datasets, 'get_related_dataset': get_related_dataset} |
| |
|
| |
|