nigel.stanger / Digital_Repository
Transfer to URL with SHA Find file
Newer
Older
Digital_Repository / Repositories / Misc / import_users.pl
nstanger on 23 Mar 2010 2 KB - Added bulk user import script.
Raw Blame History 
  1. #!/usr/bin/perl -w -I/home/eprints3/perl_lib
  2.  
  3. use EPrints::DataObj::User;
  4. use EPrints::Session;
  5. use strict;
  6.  
  7. my $session = EPrints::Session->new( 1 , $ARGV[0] );
  8. exit unless( defined $session );
  9.  
  10. # Set to 1 to actually make changes to the database
  11. my $forreal = 1;
  12.  
  13. # Get user dataset
  14. my $user_ds = $session->get_archive()->get_dataset( "user" );
  15.  
  16.  
  17. my $input_filename = $ARGV[1];
  18.  
  19. ( defined $input_filename ) or die "No input filename specified.\n";
  20.  
  21. # Try to open the input file.
  22. open INFILE, "<$input_filename"
  23. 	or die "ERROR: couldn't open file '$input_filename' for reading.\n";
  24.  
  25.  
  26. while( <INFILE> )
  27. {
  28.  
  29. 	next if /^#/;
  30. 	next if /^$/;
  31. 	
  32. 	#username	type	password	given name	family name	email	department	organisation	address	country	homepage
  33. 	/^([^\t]+)\t([^\t]+)\t([^\t]+)\t([^\t]+)\t([^\t]+)\t([^\t]+)\t([^\t]+)\t([^\t]+)\t"?([^"]+)"?\t([^\t]+)\t(.*)$/;
  34.  
  35. 	my $username = $1;
  36. 	my $usertype = $2;
  37. 	my $password = EPrints::Utils::crypt_password( $3 );
  38. 	my $givenname = $4;
  39. 	my $familyname = $5;
  40. 	my $email = $6;
  41. 	my $department = $7;
  42. 	my $organisation = $8;
  43. 	my $address = $9;
  44. 	my $country = $10;
  45. 	my $homepage = $11;
  46. 	
  47. 	# Does account already exist?
  48. 	my $searchexp = new EPrints::SearchExpression( session=>$session, dataset=>$user_ds );
  49. 	$searchexp->add_field( $user_ds->get_field( "username" ), $username );
  50. 	my $searchid = $searchexp->perform_search;
  51. 	my( $user ) = $searchexp->get_records;
  52. 	$searchexp->dispose( );
  53.  
  54. 	if( !defined $user )
  55. 	{
  56.  
  57. 		# New account
  58. 		if( $forreal )
  59. 		{
  60. 			$user = EPrints::User::create( $session, $usertype );
  61. 			$user->set_value( "username", $username );
  62. 			print "CREATING: $username\n";
  63. 		}
  64. 		else
  65. 		{
  66. 			print "(WOULD BE) CREATING: $username\n";
  67. 		}
  68.  
  69. 	} 
  70. 	else 
  71. 	{
  72.  
  73. 		# Update account
  74. 		if( $forreal )
  75. 		{
  76. 			print "UPDATING: $username\n";
  77. 		} 
  78. 		else 
  79. 		{
  80. 			print "(WOULD BE) UPDATING: $username\n";
  81. 		}
  82.  
  83. 	}
  84.  
  85. 	# Set metadata
  86. 	if( $forreal )
  87. 	{
  88.  
  89. 		my $name = {};
  90. 		$name->{family} = $familyname;
  91. 		$name->{given} = $givenname;
  92. 		$user->set_value( "name", $name );
  93. 		$user->set_value( "email", $email );
  94. 		$user->set_value( "password", $password );
  95. 		$user->set_value( "dept", $department );
  96. 		$user->set_value( "org", $organisation );
  97. 		$user->set_value( "address", $address );
  98. 		$user->set_value( "country", $country );
  99. 		$user->set_value( "url", $homepage );
  100. 		
  101. 		$user->commit();
  102.  
  103. 	} 
  104. 	else 
  105. 	{
  106.  
  107. 		print "FAMILY = $familyname\n";
  108. 		print "GIVEN = $givenname\n";
  109. 		print "EMAIL = $email\n";
  110. 		print "PASSWORD = $password\n";
  111. 		print "DEPARTMENT = $department\n";
  112. 		print "ORGANISATION = $organisation\n";
  113. 		print "ADDRESS = $address\n";
  114. 		print "COUNTRY = $country\n";
  115. 		print "URL = $homepage\n\n";
  116.  
  117. 	}
  118.  
  119. }
  120.  
  121. # Clean up.
  122. close INFILE;